Update app.py

app.py

@@ -18,7 +18,7 @@ def generate_text(prompt, length=100, theme=None, **kwargs):
     generation_config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/generation_config.json"
     
     # Load the model from the Hugging Face space
-    model = transformers.GPT2LMHeadModel.from_pretrained("./pytorch_model.bin", config="./config.json").to(device)
+    model = transformers.GPT2LMHeadModel.from_pretrained(model_url, config=config_url).to(device)
     
     # Load the tokenizer from the Hugging Face space
     tokenizer = transformers.GPT2Tokenizer.from_pretrained('gpt2')