Update app.py

app.py

@@ -2,9 +2,20 @@ import tensorflow as tf
 import cv2
 import numpy as np
 from glob import glob
-from models import Yolov4
+# from models import Yolov4
 import gradio as gr
-model = Yolov4(weight_path="best.pt", class_name_path='coco_classes.txt')
+# model = Yolov4(weight_path="best.pt", class_name_path='coco_classes.txt')
+
+
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO("best.pt")  # load a custom model
+
+# Predict with the model
+# results = model("image.jpg", save = True)  # predict on an image
+
+
 def gradio_wrapper(img):
     global model
     #print(np.shape(img))