Update app.py

app.py

@@ -18,7 +18,7 @@ def AImodel(text,questionText):
     
 
     # a) Get predictions
-    nlp = pipeline('text-generation', model=model_name, tokenizer=model_name)
+    nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
     QA_input = {
         'question': questionText,
         'context': text