Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import torch
 import pytorch_lightning as pl
 from pytorch_lightning.callbacks.early_stopping import EarlyStopping
 from transformers import (
@@ -40,7 +41,7 @@ def predict(text):
             )
             for g in generated_ids
         ]
-    return ['Q: ' + text for text in preds]
+    return preds
 
 # text_to_predict = predict(text)
 # predicted = ['Q: ' + text for text in predict(text_to_predict)]