emielclopterop commited on
Commit
d309eb4
·
verified ·
1 Parent(s): 07db68b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -3,14 +3,18 @@
3
  import gradio as gr
4
  from transformers import pipeline
5
 
6
- #pipelines
7
  qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
 
8
  classification_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
9
  translation_pipeline = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
 
10
  topic_classification_pipeline = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english") # Fine-tuned model for topic classification
 
11
  summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
12
 
13
- #functions
14
 
15
  def answer_question(context, question):
16
  return qa_pipeline(question=question, context=context)["answer"]
@@ -42,7 +46,7 @@ def multi_model_interaction(text):
42
  }
43
 
44
 
45
-
46
  with gr.Blocks() as demo:
47
  with gr.Tab("Single Models"):
48
  with gr.Column():
@@ -83,4 +87,5 @@ with gr.Blocks() as demo:
83
  multi_output = gr.Text(label="Results")
84
  gr.Button("Process").click(multi_model_interaction, inputs=[input_text], outputs=multi_output)
85
 
 
86
  demo.launch(share=True, debug=True)
 
3
  import gradio as gr
4
  from transformers import pipeline
5
 
6
+ #pipelines init
7
  qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
8
+
9
  classification_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
10
+
11
  translation_pipeline = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
12
+
13
  topic_classification_pipeline = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english") # Fine-tuned model for topic classification
14
+
15
  summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
16
 
17
+ #functions defining
18
 
19
  def answer_question(context, question):
20
  return qa_pipeline(question=question, context=context)["answer"]
 
46
  }
47
 
48
 
49
+ #Blocking interface
50
  with gr.Blocks() as demo:
51
  with gr.Tab("Single Models"):
52
  with gr.Column():
 
87
  multi_output = gr.Text(label="Results")
88
  gr.Button("Process").click(multi_model_interaction, inputs=[input_text], outputs=multi_output)
89
 
90
+ #Launching demo
91
  demo.launch(share=True, debug=True)