Update app.py

app.py

@@ -4,13 +4,10 @@ from generic import save_image_vis
 
 
 def generate_viz(URL, question):
-    
     print(f"Input Image URL: {URL}")
     print(f"Target Class: {question}")
-
-    img, html_page, answer = save_image_vis(URL, question)
-    
-    return img, answer, html_page
+    oring, img, html_page, answer = save_image_vis(URL, question)
+    return answer, oring, img, html_page
 
 title = "Explain LXMERT 🤯"
 
@@ -19,7 +16,10 @@ iface = gr.Interface(fn=generate_viz, inputs=[
         gr.Text(label="Input Image URL"),
         gr.Text(label="Input the Question"),
     ],
-    outputs=[ gr.Image(label="EXplain Image"), gr.Text(label="Answer"), gr.HTML(label="Visualize Sentence Importance"),],
+    outputs=[gr.Text(label="Answer"),
+                              gr.Image(label="Oringinal Image", width=450, height=300),
+                              gr.Image(label="EXplain Image", width=450, height=300),
+                              gr.HTML(label="Visualize Sentence Importance"), ],
     title=title,
     allow_flagging="never",
     cache_examples=True,