Spaces:

sailormars18
/

Yelp-reviews-usingGPT2

Runtime error

sailormars18 commited on Jun 8, 2023

Commit

7b10293

1 Parent(s): 2672584

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # 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_name = "sailormars18/Yelp-reviews-usingGPT2"
     # Load the model from the Hugging Face space
     model = GPT2LMHeadModel.from_pretrained(model_name).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_name = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/tree/main"
     # Load the model from the Hugging Face space
     model = GPT2LMHeadModel.from_pretrained(model_name).to(device)