Update app.py

app.py

@@ -5,9 +5,6 @@ from ultralyticsplus import YOLO, render_result
 import cv2
 import numpy as np
 
-model = YOLO('best (1).pt')
-
-
 # def response(image):
 #   print(image)
 #   results = model(image)
@@ -21,8 +18,10 @@ model = YOLO('best (1).pt')
       
 #       return im_rgb
 
-def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
-    
+def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6
+             
+    model = YOLO('best (1).pt')
+
     results = model.predict(image, conf=conf_threshold, iou=iou_threshold, imgsz=image_size)
 
     box = results[0].boxes