Update app.py

app.py

@@ -1,9 +1,16 @@
-import json
+import gradio as gr
+from transformers import pipeline
 
-# Load the dataset (update 'your_file_path.json' with the actual file location)
-file_path = "https://huggingface.co/spaces/BINAII/Mini_Medical_Assistant/blob/main/release_conditions.json"
-with open(file_path, "r") as file:
-    data = json.load(file)
+model_name = "emilyalsentzer/Bio_ClinicalBERT"
+nlp_pipeline = pipeline("fill-mask", model=model_name)
 
-# Display a sample record
-print(data[0])  # Print the first patient record
+def medical_chatbot(user_input):
+    if "[MASK]" not in user_input:
+        user_input += " [MASK]"
+    
+    response = nlp_pipeline(user_input)
+    return {"prediction": response[0]["sequence"], "confidence": response[0]["score"]}
+
+# Create an API interface
+interface = gr.Interface(fn=medical_chatbot, inputs="text", outputs="json")
+interface.launch(share=True)