Update app.py

app.py

@@ -38,7 +38,7 @@ def predict(text):
         for g in generated_ids
     ]
     
-    output = ['Q: ' + text for text in preds]
+    output = ['Question: '+ text.replace('A', 'Answer') for text in preds]
     final_str = '\n'.join([f"{i+1}. {s}" for i, s in enumerate(output)])
     #final_str = '\n'.join([f"{i+1}. Question: {s.split('Answer')[0].strip()}\n   Answer{s.split('Answer')[1].strip()}" for i, s in enumerate(output)])