Update app.py

app.py

@@ -4,8 +4,8 @@ import re
 
 # Load the fine-tuned model and tokenizer
 try:
-    model = GPT2LMHeadModel.from_pretrained("Manasa1/finetuned_GPTb")  # Path to your fine-tuned GPT-2 model
-    tokenizer = GPT2Tokenizer.from_pretrained("Manasa1/finetuned_GPTb")  # Path to tokenizer
+    model = GPT2LMHeadModel.from_pretrained("Manasa1/finetuned_distillGPT2")  # Path to your fine-tuned GPT-2 model
+    tokenizer = GPT2Tokenizer.from_pretrained("Manasa1/finetuned_distillGPT2")  # Path to tokenizer
     tokenizer.pad_token = tokenizer.eos_token  # Ensure pad_token is set correctly
 except Exception as e:
     print(f"Error loading model or tokenizer: {e}")
@@ -27,7 +27,7 @@ def generate_answer(question):
             no_repeat_ngram_size=2,
             top_p=0.9,
             top_k=50,
-            temperature=0.7,
+            temperature=0.8,
             do_sample=True,
             pad_token_id=tokenizer.eos_token_id
         )
@@ -75,4 +75,3 @@ with gr.Blocks() as app:
 # Run the app
 if __name__ == "__main__":
     app.launch()
-