Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a more suitable model for medical diagnosis (BioBERT)
-model = pipeline("zero-shot-classification", model="dmis-lab/biobert-base-cased-v1.1")
+# Use a zero-shot classification model
+model = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
 # Candidate labels for medical diagnosis
 candidate_labels = [