Spaces:

sailormars18
/

Yelp-reviews-usingGPT2

Runtime error

sailormars18 commited on Apr 13, 2023

Commit

bcb1530

1 Parent(s): b8392d0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import transformers
 # Define a function for generating text based on a prompt using the fine-tuned GPT-2 model and the tokenizer
 def generate_text(prompt, length=100, theme=None, **kwargs):
-    model_url = "https://huggingface.co/sailormars18/Yelp-reviews-usingGPT2/pytorch_model.bin"
     # Load the model from the Hugging Face space
     model = transformers.GPT2LMHeadModel.from_pretrained(model_url).to(device)

 # Define a function for generating text based on a prompt using the fine-tuned GPT-2 model and the tokenizer
 def generate_text(prompt, length=100, theme=None, **kwargs):
+    model_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/pytorch_model.bin"
     # Load the model from the Hugging Face space
     model = transformers.GPT2LMHeadModel.from_pretrained(model_url).to(device)