Update app.py

app.py

@@ -31,18 +31,10 @@ def askme(symptoms, question):
     outputs = model.generate(**inputs, max_new_tokens=200, use_cache=True)
     response_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0].strip() #skip_special_tokens=True
     # Remove system messages and content
-    # Extract only the assistant's response
-    #assistant_response = response_text.split("<|im_start|>assistant")[1].strip().replace('<|im_end|>', '')
    #  Extract only the assistant's response
     assistant_response =response_text.split("assistant")[1].strip().split("user")[0].strip()   
     return assistant_response
     
-    
-    
-    
-    
-    
-
 # Example usage
 symptoms = '''\
 I'm a 35-year-old male and for the past few months, I've been experiencing fatigue,
@@ -59,7 +51,7 @@ examples = [
 
 iface = gr.Interface(
     fn=askme,
-    inputs=["text", "text"],
+    inputs=["Symptoms", "Question"],
     outputs="text",
     examples=examples,
     title="Medical AI Chatbot",