quocviethere
commited on
Commit
•
213f335
1
Parent(s):
f7f5995
Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,30 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
+
from transformers import pipeline
|
3 |
|
4 |
+
# Load the model using Hugging Face's pipeline
|
5 |
+
model_name = "quocviethere/xlmroberta-finetuned-squadv2")"
|
6 |
+
qa_pipeline = pipeline('question-answering', model=model_name)
|
7 |
+
|
8 |
+
# Define a function that the interface will use
|
9 |
+
def answer_question(context, question):
|
10 |
+
return qa_pipeline(context=context, question=question)['answer']
|
11 |
+
|
12 |
+
# Define a single example
|
13 |
+
examples = [
|
14 |
+
["Tính đến tháng 6/2023, Vườn ươm Viện Đổi mới sáng tạo đã ươm tạo gần 100 dự án khởi nghiệp, trong đó có 19 dự án thành lập công ty, hơn 7600 người tham dự các hoạt động và tổng giá trị đầu tư mạo hiểm đạt 350.000 USD.",
|
15 |
+
"Tổng giá trị đầu tư mạo hiểm của Vườn ươm Viện Đổi mới sáng tạo là bao nhiêu?"]
|
16 |
+
]
|
17 |
+
|
18 |
+
# Create the Gradio interface
|
19 |
+
iface = gr.Interface(fn=answer_question,
|
20 |
+
inputs=[
|
21 |
+
gr.Textbox(lines=7, placeholder="Enter the context here...", label="Context"),
|
22 |
+
gr.Textbox(placeholder="Enter your question here...", label="Question")
|
23 |
+
],
|
24 |
+
outputs=gr.Textbox(label="Answer"),
|
25 |
+
title='UEH-QA',
|
26 |
+
description='Creating safe AGI that benefits all of humanity.',
|
27 |
+
examples=examples)
|
28 |
+
|
29 |
+
# Launch the interface
|
30 |
+
iface.launch()
|