Update app.py

app.py

@@ -78,7 +78,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=True):
     )
 
     if raw_text:
-        return None, data[0]['generated_text'][0]
+        return None, data
 
     width, height = 3326, 3326
     assets_path = "assets"