Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import cv2
+from ultralytics import YOLO
 #!pip install  transformers
 #from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
@@ -7,7 +8,7 @@ import torch
 model_path = "best.pt"
 #tokenizer = AutoTokenizer.from_pretrained(model)
 #model = AutoModelForSequenceClassification.from_pretrained(model)
-loaded_model = torch.load(model_path, map_location=torch.device('cpu'))
+loaded_model = YOLO(model_path)
 # Define the Gradio Interface
 import easyocr
 def image_classifier(img):