Update app.py

app.py

@@ -1,32 +1,35 @@
-
 import gradio as gr
-from transformers import pipeline
-from transformers import SummarizationPipeline
-
-#summarizer = SummarizationPipeline(model="hamzamalik11/Biobart_radiology_summarization", task="summarization", tokenizer="hamzamalik11/Biobart_radiology_summarization")
-
-pipeline = pipeline(task="summarization", model="hamzamalik11/Biobart_radiology_summarization")
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-def greet(RADIOLOGY_REPORT):
-     summary_output= pipeline(RADIOLOGY_REPORT)
-     return summary_output
+model_checkpoint = "hamzamalik11/Biobart_radiology_summarization"  
+model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
+tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
-findings_examples = [
-    "prevoid bladder volume cc postvoid bladder volume cc bladder grossly normal appearance",
-    "heart mediastinal contours normal left sided subclavian line position tip distal svc lungs remain clear active disease effusions",
-    '''heart size normal mediastinal hilar contours remain stable small right pneumothorax remains unchanged surgical lung staples overlying
-    left upper lobe seen linear pattern consistent prior upper lobe resection soft tissue osseous structures appear unremarkable nasogastric
-    endotracheal tubes remain satisfactory position atelectatic changes right lower lung field remain unchanged prior study''',
-]
-
-radiology_demo = gr.Interface(fn=greet, inputs="text", outputs="text",
-    examples=findings_examples,
-    description="""                                                                -------------RADIOLOGY REPORT SUMMARIZATION----------------
-                                                    ENTER YOUR FINDINGS IN RADIOLOGY REPORT SECTION  &  MODEL WILL PROVIDE YOU IMPRESSIONS AS AN OUTPUT""",
-
-)
+from transformers import SummarizationPipeline
+summarizer = SummarizationPipeline(model=model, tokenizer=tokenizer)
 
-demo = gr.TabbedInterface([radiology_demo])
+import gradio as gr
 
+examples = [ "heart mediastinal contours normal left sided subclavian line position tip distal svc lungs remain clear active disease effusions",
+  "prevoid bladder volume cc postvoid bladder volume cc bladder grossly normal appearance"
+ ]
+
+description = """
+THIS MODEL SUMMARIZE FINDINGS OF RADIOLOGY REPORTS INTO IMPRESSIONS 
+<b>Enter a findings of radiology report to see the generated impression!</b>
+"""
+
+def summarize(radiology_report):
+  summary = summarizer(radiology_report)[0]['summary_text']
+  return summary
+
+iface = gr.Interface(fn=summarize, 
+             inputs=gr.inputs.Textbox(lines=5, label="Radiology Report"),
+             outputs=gr.outputs.Textbox(label="Summary"),
+             examples=examples,
+             title="Radiology Report Summarization",
+             description=description,
+             theme="huggingface")
+             
 if __name__ == "__main__":
-    demo.launch( share = True)
+    iface.launch(share=False)