Create app.py

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+def number_detector(s):
+    model = YOLO("best.pt")
+    res = model.predict(source=s,conf=0.25)
+    box1 = res[0].boxes.xywh[0]
+    bounding_box1 = box1.cpu().NumPy()
+    box2 = res[0].boxes.xywh[1]
+    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()
+    number=(digit1*10)+digit2
+    
+    return number
+demo = gr.Interface(
+    fn=number_detector,
+    inputs=gr.Image(type="numpy"),
+    outputs=gr.Textbox())
+
+demo.launch(debug = True ,share = True)
+
+# Launch the Gradio Interface
+demo.launch()