Update app.py

app.py

@@ -6,11 +6,14 @@ bio_gpt = pipeline("text-generation", model="microsoft/biogpt")
 
 def medical_chatbot(query):
     # Generate text using BioGPT
-    result = bio_gpt(query, max_length=100, num_return_sequences=1)
+    result = bio_gpt(query, max_length=1000, num_return_sequences=1)
 
     # 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
@@ -21,4 +24,4 @@ iface = gr.Interface(
     title="Medical Chatbot"
 )
 
-iface.launch()
+iface.launch()