Update app.py

app.py

@@ -10,20 +10,20 @@ tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
 model = BertForQuestionAnswering.from_pretrained("CountingMstar/ai-tutor-bert-model")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-# def get_prediction(context, question):
-#   inputs = tokenizer.encode_plus(question, context, return_tensors='pt').to(device)
-#   outputs = model(**inputs)
+def get_prediction(context, question):
+  inputs = tokenizer.encode_plus(question, context, return_tensors='pt').to(device)
+  outputs = model(**inputs)
   
-#   answer_start = torch.argmax(outputs[0])  
-#   answer_end = torch.argmax(outputs[1]) + 1 
+  answer_start = torch.argmax(outputs[0])  
+  answer_end = torch.argmax(outputs[1]) + 1 
   
-#   answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs['input_ids'][0][answer_start:answer_end]))
+  answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs['input_ids'][0][answer_start:answer_end]))
   
-#   return answer
+  return answer
   
-# def question_answer(context, question):
-#   prediction = get_prediction(context,question)
-#   return prediction
+def question_answer(context, question):
+  prediction = get_prediction(context,question)
+  return prediction
 
 def split(text):
     context, question = '', ''
@@ -44,14 +44,14 @@ def split(text):
         
     return context[:-2], question[1:]
     
-# def greet(texts):
-#     context, question = split(texts)
-#     answer = question_answer(context, question)
-#     return answer
-def greet(text):
-    context, question = split(text)
-    # answer = question_answer(context, question)
-    return context
+def greet(texts):
+    context, question = split(texts)
+    answer = question_answer(context, question)
+    return answer
+# def greet(text):
+#     context, question = split(text)
+#     # answer = question_answer(context, question)
+#     return context
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()