Update app.py

app.py

@@ -200,20 +200,12 @@ def inference(video, lang, full_scan, number_filter, use_trocr, time_step, perio
     buf = io.BytesIO()
     plot_fig.savefig(buf, format='png')
     buf.seek(0)
-    plot_image = PIL.Image.open(buf)
-
-    # Resize the image to fit the width of the returned image
-    im_width, im_height = im.size
-    plot_width, plot_height = plot_image.size
-    new_plot_height = int(plot_height * im_width / plot_width)
-    resized_plot_image = plot_image.resize((im_width, new_plot_height), PIL.Image.ANTIALIAS)
-
     # Convert the resized image to a NumPy array
-    plot_np = np.array(resized_plot_image)
+    plot_np = np.array(PIL.Image.open(buf))
 
     # Close the buffer
     buf.close()
-    
+
     return output, im, plot_np, df  # Change this line
 
 
@@ -252,9 +244,9 @@ gr.Interface(
     ],
     [
         gr.outputs.Video(label='Output Video'),
-        gr.outputs.Image(label='Output Preview', type='numpy'),
+        gr.outputs.Image(label='Output Preview', type='numpy').style(height=480),
         # gr.Plot(label='Temporal Profile'),
-        gr.outputs.Image(label='Temporal Profile', type='numpy'),
+        gr.outputs.Image(label='Temporal Profile', type='numpy').style(height=640),
         gr.outputs.Dataframe(headers=['Box', 'Time (s)', 'Text'], type='pandas')
     ],
     title=title,