Update app.py

app.py

@@ -21,12 +21,30 @@ def get_prediction(context, question):
 def question_answer(context, question):
   prediction = get_prediction(context,question)
   return prediction
+
+def split(texts):
+    words = word_tokenize(texts)
+    context, question = '', ''
+    act = False
+
+    for w in words:
+        if w == '///':
+            act = True
+            
+        if act == False:
+            context += w + ' '
+        else:
+            if w == '///':
+                w = ''
+            question += w + ' '
+
+    context = context[:-1]
+    question = question[1:-1]
+    return context, question
     
 def greet(texts):
-    question = texts[:int(len(texts)/2)]
-    answer = texts[int(len(texts)/2):]
-    # for question, answer in texts:
-    #     question_answer(context, question)
+    context, question = split(texts)
+    answer = question_answer(context, question)
     return answer
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")