Spaces:

sailormars18
/

Yelp-reviews-usingGPT2

Runtime error

sailormars18 commited on Jun 8, 2023

Commit

f07e8e2

1 Parent(s): fa65b7a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,8 @@ 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_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/resolve/main/pytorch_model.bin"
-    config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/resolve/main/config.json"
     # Download the model and configuration files
     model_path = "./pytorch_model.bin"

 # 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"
+    config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/config.json"
     # Download the model and configuration files
     model_path = "./pytorch_model.bin"