Medora / app.py
Sid971168's picture
Update app.py
4d29da8 verified
raw
history blame contribute delete
488 Bytes
from transformers import pipeline
# AI Pipeline Load Karo
nlp = pipeline("question-answering", model="dmis-lab/biobert-base-cased-v1.1")
# Function: Symptoms Ka Answer Generate Karega
def get_medical_advice(symptoms):
response = nlp({
'question': f"Best medicine for {symptoms}?",
'context': "Provide medicine recommendations based on disease."
})
return response['answer']
# Example Call
symptoms = "fever and headache"
print(get_medical_advice(symptoms))