Update app.py

app.py

@@ -14,16 +14,14 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 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_name = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/config.json"
-    generation_config_name = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/generation_config.json"
+    config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/config.json"
+    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(model_url, config_name=config_name, 
-                                                          generation_config_name=generation_config_name).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(model_url, config_name=config_name, 
-                                                            generation_config_name=generation_config_name)
+    tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_url, config=config_url)
 
     # If a theme is specified, add it to the prompt as a prefix for a special token
     if theme: