quocviethere commited on
Commit
213f335
1 Parent(s): f7f5995

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -1
app.py CHANGED
@@ -1,3 +1,30 @@
1
  import gradio as gr
 
2
 
3
- gr.load("models/quocviethere/xlmroberta-finetuned-squadv2").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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()