Update app.py

app.py

@@ -109,7 +109,7 @@ def test_error( images, x_mask):
     while (ind<len(ss)-1):
       k=(len(ss)-2)-ind
       outstr=outstr+worddicts_r[int(ss[k])]
-      '''textimg = Image.new('RGB', (1400,100),(255,255,255))
+      textimg = Image.new('RGB', (1400,100),(255,255,255))
       drawtext = ImageDraw.Draw(textimg)
       drawtext.text((20, 20), outstr ,(0,0,0),font=font)
       fig,axes=plt.subplots(2,1)
@@ -121,13 +121,13 @@ def test_error( images, x_mask):
       axes[1].imshow(255-(255 * visualization), alpha=0.2)
       plt.axis('off')
 
-      plt.savefig('/content/gdrive/My Drive/CALText_Demo/res.png')
-      frames.append(Image.fromarray(cv2.imread('/content/gdrive/My Drive/CALText_Demo/res.png'), 'RGB'))'''
+      plt.savefig('res.png')
+      frames.append(Image.fromarray(cv2.imread('res.png'), 'RGB'))
       ind=ind+1
-    '''frame_one = frames[0]
-    frame_one.save("/content/gdrive/My Drive/CALText_Demo/'vis.gif", format="GIF", append_images=frames,save_all=True, duration=300, loop=0)
-    gif_image="/content/gdrive/My Drive/CALText_Demo/'vis.gif"'''
-  return outstr
+    frame_one = frames[0]
+    frame_one.save("vis.gif", format="GIF", append_images=frames,save_all=True, duration=300, loop=0)
+    gif_image="vis.gif"
+  return outstr,gif_image
 
 
 
@@ -155,7 +155,7 @@ article = "<p style='text-align: center'></p>"
 css = "#0 {object-fit: contain;} #1 {object-fit: contain;}"
 inputs = gr.inputs.Image(label="Input Image")
 
-demo = gr.Interface(fn=recognize_text,inputs=inputs,outputs=[gr.Textbox(label="Output")],title=title,
+demo = gr.Interface(fn=recognize_text,inputs=inputs,outputs=[gr.Textbox(label="Output"), gr.Image(label="Attended Regions")],title=title,
                     description=description,
                     article=article,allow_flagging='never')