Deploy the trained model

inference.py

@@ -0,0 +1,19 @@
+from transformers import pipeline
+import string
+
+# Replace this with your own checkpoint
+model_checkpoint = "results/checkpoint-16000"
+question_answerer = pipeline("question-answering", model = model_checkpoint)
+
+def predict(question, context):
+    answer = question_answerer(question = question,
+                               context  = context)
+    
+    exclude = set(string.punctuation)
+    
+    text = answer['answer']
+    text = ''.join(ch for ch in text if ch not in exclude)
+    
+    answer['answer'] = text
+
+    return  answer