hamzamalik11 commited on
Commit
5197153
·
1 Parent(s): 42f1ba1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,12 +2,12 @@ import gradio as gr
2
  from transformers import pipeline
3
  from transformers import SummarizationPipeline
4
 
5
- summarizer = SummarizationPipeline(model="hamzamalik11/Biobart_radiology_summarization", task="summarization", tokenizer="hamzamalik11/Biobart_radiology_summarization")
6
 
7
- #pipeline = pipeline(, model="hamzamalik11/Biobart_radiology_summarization")
8
 
9
  def greet(RADIOLOGY_REPORT):
10
- summary_output= summarizer(RADIOLOGY_REPORT)
11
  return summary_output
12
 
13
  findings_examples = [
 
2
  from transformers import pipeline
3
  from transformers import SummarizationPipeline
4
 
5
+ #summarizer = SummarizationPipeline(model="hamzamalik11/Biobart_radiology_summarization", task="summarization", tokenizer="hamzamalik11/Biobart_radiology_summarization")
6
 
7
+ pipeline = pipeline(task="summarization", model="hamzamalik11/Biobart_radiology_summarization")
8
 
9
  def greet(RADIOLOGY_REPORT):
10
+ summary_output= pipeline(RADIOLOGY_REPORT)
11
  return summary_output
12
 
13
  findings_examples = [