Update app.py

app.py

@@ -16,12 +16,20 @@ 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_url).to(device)
+    model = transformers.GPT2LMHeadModel.from_pretrained(model_url, config=config, **generation_config).to(device)
     
     # Load the tokenizer from the Hugging Face space
-    tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_url, config=config_url)
+    tokenizer = transformers.GPT2Tokenizer.from_pretrained('gpt2')
 
     # If a theme is specified, add it to the prompt as a prefix for a special token
     if theme:
@@ -36,7 +44,6 @@ def generate_text(prompt, length=100, theme=None, **kwargs):
 
     sample_outputs = model.generate(
         input_ids,
-        config=generation_config_url,
         attention_mask=attention_mask,
         pad_token_id=pad_token_id,
         do_sample=True,