Update app.py

app.py

@@ -13,9 +13,9 @@ 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_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"
+    model_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/raw/main/pytorch_model.bin"
+    config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/raw/main/config.json"
+    generation_config_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/raw/main/generation_config.json"
     
     # Load the model from the Hugging Face space
     model = transformers.GPT2LMHeadModel.from_pretrained(model_url, config=config_url).to(device)