sehatguard commited on
Commit
dbbf677
·
verified ·
1 Parent(s): 9a96a89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Load the pre-trained model (you can replace this with any healthcare model from Hugging Face)
5
- model = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
6
 
7
  # Function to diagnose based on symptoms
8
  def diagnose(symptoms):
 
2
  from transformers import pipeline
3
 
4
  # Load the pre-trained model (you can replace this with any healthcare model from Hugging Face)
5
+ model = pipeline("fill-mask", model="medicalai/ClinicalBERT")
6
 
7
  # Function to diagnose based on symptoms
8
  def diagnose(symptoms):