Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import pipeline
-import PyTorch, TensorFlow
 
 from Gradio_UI import GradioUI
 
@@ -26,6 +25,8 @@ def generate_relations(text):
 demo = gr.Interface(
     fn=generate_relations,
     inputs=gr.Textbox(lines=10, label="Clinical Note or Drug Review Text"),
+    inputs=gr.Textbox(lines=1, label="Entity 1"),
+    inputs=gr.Textbox(lines=1, label="Entity 2"),
     outputs=gr.Textbox(label="Extracted Causal Relations"),
     title="Causal Relation Extractor with MedLlama",
     description="Paste your clinical note or drug review. This AI agent extracts drug-condition or symptom causal relations using a fine-tuned LLM.",