Update app.py

app.py

@@ -16,17 +16,9 @@ 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"
     generation_config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/generation_config.json"
-
-    # Load the model configuration from the config.json file
-    with open(config_url, 'r') as f:
-        config = json.load(f)
-    
-    # Load the generation configuration from the generation_config.json file
-    with open(generation_config_url, 'r') as f:
-        generation_config = json.load(f)
     
     # Load the model from the Hugging Face space
-    model = transformers.GPT2LMHeadModel.from_pretrained(model_url, config=config, **generation_config).to(device)
+    model = transformers.GPT2LMHeadModel.from_pretrained(model_url, config=config_url, **generation_config_url).to(device)
     
     # Load the tokenizer from the Hugging Face space
     tokenizer = transformers.GPT2Tokenizer.from_pretrained('gpt2')