Update app.py

app.py

@@ -5,16 +5,25 @@ import gradio as gr
 import torch
 import transformers
 
+import json
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
+
+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/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"
     
     # Load the model from the Hugging Face space
-    model = transformers.GPT2LMHeadModel.from_pretrained(model_url).to(device)
+    model = transformers.GPT2LMHeadModel.from_pretrained(model_url, config_name=config_name, 
+                                                          generation_config_name=generation_config_name).to(device)
     
     # Load the tokenizer from the Hugging Face space
-    tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_url)
+    tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_url, config_name=config_name, 
+                                                            generation_config_name=generation_config_name)
 
     # If a theme is specified, add it to the prompt as a prefix for a special token
     if theme: