Update app.py

app.py

@@ -56,6 +56,7 @@ def detect_objects_in_image(image):
     if isinstance(image, np.ndarray):
         print("Converting NumPy array to PIL Image")
         image = Image.fromarray(image)
+    print(image.size)
     orig_w, orig_h = image.size
 
 
@@ -106,9 +107,9 @@ def detect_objects_in_image(image):
                         (0, 255, 0), 2)
 
     # Generate and return graph instead of dictionary
-    graph_image = generate_vehicle_count_graph(object_counts)
-
-    return Image.fromarray(img_array), graph_image  # Now returning only 2 outputs
+    # graph_image = generate_vehicle_count_graph(object_counts)
+    
+    return Image.fromarray(img_array)#, graph_image  # Now returning only 2 outputs
 
 
 # def generate_vehicle_count_graph(object_counts):
@@ -179,6 +180,8 @@ def detect_objects_in_video(video_input):
         if not ret:
             break
 
+            
+
         image = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
 
         # Get frame with detected objects and graph
@@ -193,8 +196,6 @@ def detect_objects_in_video(video_input):
     return temp_video_output, graph_image  # Return both expected outputs
 
 
-def greet(name):
-    return "Hello " + name + "!!"
 
 
 demo = gr.Interface(fn=greet, inputs="text", outputs="text")