Create app.py

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+import cv2
+import numpy as np
+from ultralytics import YOLO
+
+model = YOLO('best (7).pt')
+
+def predict_image(img): 
+        
+  model = YOLO('best (6).pt')
+  result = model.predict(img,conf=0.97)
+  res_plotted = result[0].plot()
+  #cv2.imshow( res_plotted)
+  return res_plotted
+
+
+
+image = gr.inputs.Image()
+
+label = gr.outputs.Label('ok')
+gr.Interface(fn=predict_image, inputs=image, outputs=image).launch(debug='True')