Update app.py

app.py

@@ -37,14 +37,20 @@ def detect(inp):
     return hub_model.to(device)
 
 model = custom(path_or_model='best.pt')
-  
+
+def detect1(inp):
+ #g = (size / max(inp.size))  #gain
+ #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize 
+ results = model(inp,size=640)  # inference
+ results.render()  # updates results.imgs with boxes and labels
+ return Image.fromarray(results.imgs[0])  
 
 inp = gr.inputs.Image(type="filepath", label="Input")
 #output=gr.outputs.Image(type="pil", label="Output Image")
 output = gr.outputs.Image(type="filepath", label="Output")
 #.outputs.Textbox()
 
-io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='Pot Hole Detection With Custom YOLOv7 ',
+io=gr.Interface(fn=detect1, inputs=inp, outputs=output, title='Pot Hole Detection With Custom YOLOv7 ',
 #examples=[["Examples/img-300_jpg.rf.6b7b035dff1cda092ce3dc22be8d0135.jpg"]]
 )
 #,examples=["Examples/img-300_jpg.rf.6b7b035dff1cda092ce3dc22be8d0135.jpg"]