Update app.py

app.py

@@ -1,17 +1,18 @@
 import gradio as gr
+from ultralytics import YOLO
 def number_detector(s):
     model = YOLO("best_1.pt")
-    res = model.predict(source=s,conf=0.25)
+    res = model.predict(s,conf=0.25)
     box1 = res[0].boxes.xywh[0]
-    bounding_box1 = box1.cpu().NumPy()
+    bounding_box1 = box1.cpu().numpy()
     box2 = res[0].boxes.xywh[1]
-    bounding_box2 = box2.cpu().NumPy()
+    bounding_box2 = box2.cpu().numpy()
     if res[0].boxes.data[0][0].numpy()<res[0].boxes.data[1][0].numpy():
         digit1=res[0].boxes.data[0][5].numpy()
         digit2=res[0].boxes.data[1][5].numpy()
     else:
-        digit2=res[0].boxes.data[0][5].numPy()
-        digit1=res[0].boxes.data[1][5].numPy()
+        digit2=res[0].boxes.data[0][5].numpy()
+        digit1=res[0].boxes.data[1][5].numpy()
     number=(digit1*10)+digit2
     
     return number