Spaces:
Runtime error
Runtime error
Leotis
commited on
Commit
•
058dbba
1
Parent(s):
6634278
changed model to deepset/tinyroberta-squad2
Browse files- app.py +2 -3
- inference.py +8 -1
app.py
CHANGED
@@ -4,10 +4,9 @@ from inference import Question_and_Answer_System
|
|
4 |
qa = Question_and_Answer_System()
|
5 |
|
6 |
|
7 |
-
|
8 |
-
def greet(question):
|
9 |
prediction = qa.answer_question(question)
|
10 |
return prediction
|
11 |
|
12 |
-
iface = gr.Interface(fn=
|
13 |
iface.launch()
|
|
|
4 |
qa = Question_and_Answer_System()
|
5 |
|
6 |
|
7 |
+
def ask_question(question):
|
|
|
8 |
prediction = qa.answer_question(question)
|
9 |
return prediction
|
10 |
|
11 |
+
iface = gr.Interface(fn=ask_question, inputs="text", outputs="text")
|
12 |
iface.launch()
|
inference.py
CHANGED
@@ -36,9 +36,16 @@ class Question_and_Answer_System:
|
|
36 |
return retriever
|
37 |
|
38 |
def create_reader(self, document_store):
|
|
|
39 |
reader = FARMReader(
|
40 |
-
model_name_or_path="deepset/roberta-base-squad2", use_gpu=
|
41 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
return reader
|
43 |
|
44 |
def create_pipeline(self, reader, retriever):
|
|
|
36 |
return retriever
|
37 |
|
38 |
def create_reader(self, document_store):
|
39 |
+
"""
|
40 |
reader = FARMReader(
|
41 |
+
model_name_or_path="deepset/roberta-base-squad2", use_gpu=True
|
42 |
)
|
43 |
+
"""
|
44 |
+
reader = FARMReader(
|
45 |
+
model_name_or_path="deepset/tinyroberta-squad2", use_gpu=True
|
46 |
+
)
|
47 |
+
|
48 |
+
|
49 |
return reader
|
50 |
|
51 |
def create_pipeline(self, reader, retriever):
|