Upload of app and models

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+from ultralytics import YOLO
+import cv2
+
+MODELOS = [
+    "model/best.pt", "model/bestV2.pt", "model/model_merged1.pt"
+]
+
+m = 2
+
+def predict_image(img):
+    if img is None:
+        return None
+    
+    model = YOLO(MODELOS[m])
+    results = model(img)
+
+    annotated_img = results[0].plot()
+
+    annotated_img_rgb = cv2.cvtColor(annotated_img, cv2.COLOR_BGR2RGB)
+    
+    return annotated_img_rgb
+
+demo = gr.Interface(
+    fn=predict_image,
+    inputs=gr.Image(sources=["webcam"], type="numpy", label="Tirar foto"),
+    outputs=gr.Image(type="numpy", label="Resultado do modelo"),
+    title="Detecção de Objetos com YOLO",
+    description="Clique no ícone de câmera para tirar uma foto e o modelo fará a detecção automaticamente!",
+    live=True
+)
+
+if __name__ == "__main__":
+    demo.launch()

model/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59a9095ad5630a083fc24082af5a963afa60af8c94c2637126f965fd33b24e6e
+size 5451994

model/bestV2.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b28ea4d9afeccf42d0ae9c298d8eb1dbd979e37fcb8844689dc2d754525b4349
+size 5476186

model/model_merged1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:60c77ebffcfe94e80eee4138f55ea13ec425d216b7bddec026b5cbccbbf041d1
+size 5459546

requirements.txt

@@ -0,0 +1,4 @@
+ultralytics
+gradio
+numpy
+opencv-python