Update app.py

app.py

@@ -195,16 +195,16 @@ def inference(video, lang, full_scan, number_filter, use_trocr, time_step, perio
         df_list.append({"Box": f"", "Time (s)": "", "Text": ""})
     df = pd.concat([pd.DataFrame(df_list)])
 
-    # Convert the Matplotlib plot to a NumPy array
-    plot_fig = plot_temporal_profile(temporal_profiles)
-    buf = io.BytesIO()
-    plot_fig.savefig(buf, format='png')
-    buf.seek(0)
-    # Convert the resized image to a NumPy array
-    plot_np = np.array(PIL.Image.open(buf))
+    # # Convert the Matplotlib plot to a NumPy array
+    # plot_fig = plot_temporal_profile(temporal_profiles)
+    # buf = io.BytesIO()
+    # plot_fig.savefig(buf, format='png')
+    # buf.seek(0)
+    # # Convert the resized image to a NumPy array
+    # plot_np = np.array(PIL.Image.open(buf))
 
-    # Close the buffer
-    buf.close()
+    # # Close the buffer
+    # buf.close()
 
     # return output, im, plot_np, df  # Change this line
     return output, im, im, df  # Change this line
@@ -245,9 +245,9 @@ gr.Interface(
     ],
     [
         gr.outputs.Video(label='Output Video'),
-        gr.outputs.Image(label='Output Preview', type='numpy').style(height=480),
+        gr.outputs.Image(label='Output Preview', type='numpy')#.style(height=480),
         # gr.Plot(label='Temporal Profile'),
-        gr.outputs.Image(label='Temporal Profile', type='numpy').style(height=640),
+        gr.outputs.Image(label='Temporal Profile', type='numpy')#.style(height=640),
         gr.outputs.Dataframe(headers=['Box', 'Time (s)', 'Text'], type='pandas')
     ],
     title=title,