Update app.py

app.py

@@ -16,6 +16,7 @@ tokenizer = MT5TokenizerFast.from_pretrained(
 )
 
 def predict(text):
+    
 # with torch.no_grad():
     input_ids = tokenizer.encode(text, return_tensors="pt", add_special_tokens=True)
     generated_ids = model.generate(
@@ -41,7 +42,7 @@ def predict(text):
 
     output = ['Q: ' + text for text in preds]
     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)])
-    return final_str
+   return final_str
 
 # text_to_predict = predict(text)
 # predicted = ['Q: ' + text for text in predict(text_to_predict)]