Update app.py

app.py

@@ -200,24 +200,35 @@
 # iface = gr.Interface(fn=inference, inputs="webcam", outputs="image")
 # iface.launch()
 
+import gradio as gr
 import cv2
 from ultralytics import YOLO
 
 model = YOLO('Model_IV.pt')
 print(model.names)
-webcamera = cv2.VideoCapture(0)
+# webcamera = cv2.VideoCapture(0)
 # webcamera.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
 # webcamera.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)
 
-while True:
-    success, frame = webcamera.read()
+# while True:
+#     success, frame = webcamera.read()
     
-    results = model.track(frame, conf=0.2, imgsz=480)
-    cv2.putText(frame, f"Total: {len(results[0].boxes)}", (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
-    cv2.imshow("Live Camera", results[0].plot())
+#     results = model.track(frame, conf=0.2, imgsz=480)
+#     cv2.putText(frame, f"Total: {len(results[0].boxes)}", (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
+#     cv2.imshow("Live Camera", results[0].plot())
 
-    if cv2.waitKey(1) == ord('q'):
-        break
+#     if cv2.waitKey(1) == ord('q'):
+#         break
 
-webcamera.release()
-cv2.destroyAllWindows()
+def run_model(img):
+    results = model(img)
+    annotated_img = results.render()[0]
+    return annotated_img
+
+# webcamera.release()
+# cv2.destroyAllWindows()
+
+
+
+iface = gr.Interface(fn=run_model, inputs="webcam", outputs=annotated_img)
+iface.launch()