Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|