Spaces:
Runtime error
Runtime error
import gradio as gr | |
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() |