How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="pgfeldman/Yelp_multi_cuisine")
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("pgfeldman/Yelp_multi_cuisine")
model = AutoModelForCausalLM.from_pretrained("pgfeldman/Yelp_multi_cuisine")
Quick Links

A GPT-2 model finetuned on Yelp reviews downloaded from the Yelp Open Dataset. The data from this and subsamples of cuisines are used in the paper Polling Latent Opinions: A Method for Computational Sociolinguistics Using Transformer Language Models

The model is trained to produce text in response to the following prompts:

  • "name:"
  • "review:"
  • "stars:"
Downloads last month
6
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for pgfeldman/Yelp_multi_cuisine