v8

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import AutoModelForObjectDetection
-import torch
-model_name = "best.pt"
-model = AutoModelForObjectDetection.from_pretrained(model_name)
-def predict(input_image):
+from ultralytics import YOLO
+# import torch
+model = YOLO('best.pt')
+
+    predict(input_image):
     """
     Predict model output
     """

requirement,txt

@@ -1,2 +1,2 @@
 gradio
-transformers==4.15.0
+ultralytics