Update app.py

app.py

@@ -108,7 +108,7 @@ def detect_objects_in_image(image):
 
     # Generate and return graph instead of dictionary
     # graph_image = generate_vehicle_count_graph(object_counts)
-    
+    print(Image.fromarray(img_array))
     return Image.fromarray(img_array)#, graph_image  # Now returning only 2 outputs
 
 
@@ -158,7 +158,7 @@ def generate_vehicle_count_graph(object_counts):
     buf.seek(0)
 
     plt.close(fig)  # ✅ CLOSE THE FIGURE TO FREE MEMORY
-
+    print("checkl1")
     return Image.open(buf)
 
 
@@ -192,6 +192,7 @@ def detect_objects_in_video(video_input):
 
     cap.release()
     out.release()
+    print("checl2")
 
     return temp_video_output, graph_image  # Return both expected outputs
 
@@ -216,6 +217,7 @@ title = "SpecLab Demo"
 description = "<p style='text-align: center'>Gradio demo for an ASPP model architecture trained on the SpecLab dataset. To use it, simply add your image, or click one of the examples to load them. Since this demo is run on CPU only, please allow additional time for processing. </p>"
 article = "<p style='text-align: center'><a href='https://github.com/Nano1337/SpecLab'>Github Repo</a></p>"
 css = "#0 {object-fit: contain;} #1 {object-fit: contain;}"
+print("chek3")
 demo = gr.Interface(fn=detect_objects_in_image,
                     title=title,
                     description=description,