Update app.py

app.py

@@ -83,7 +83,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
         init_font_size=150,
         right_align=False,
     )
-    return img, data['generated_text'][0]
+    return img, data['generated_text']
 
 
 gr.Interface(