Leotis commited on
Commit
058dbba
1 Parent(s): 6634278

changed model to deepset/tinyroberta-squad2

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. 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=greet, inputs="text", outputs="text")
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=False
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):