Update app.py

app.py

@@ -1,22 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load BioGPT model
 bio_gpt = pipeline("text-generation", model="microsoft/biogpt")
 
 def medical_chatbot(query):
-    # Generate text using BioGPT
-    result = bio_gpt(query, max_length=1000, num_return_sequences=1)
+    try:
+        result = bio_gpt(query, max_length=200, num_return_sequences=1, temperature=0.7, top_k=50)
+        generated_text = result[0]["generated_text"][len(query):].strip()
+        return generated_text
+    except Exception as e:
+        print(f"Error: {e}")
+        return "An error occurred."
 
-    # Extract only the generated text from response
-    generated_text = result[0]["generated_text"]
-
-    # Remove the input prompt from the generated text
-    generated_text = generated_text[len(query):].strip() #strip() to remove leading spaces
-
-    return generated_text
-
-# Create Gradio interface
 iface = gr.Interface(
     fn=medical_chatbot,
     inputs=gr.Textbox(placeholder="Ask me a medical question..."),