Update app.py

app.py

@@ -20,13 +20,15 @@ def find(name, path):
 
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt', force_reload=True)    
 def detect(inp):
- output=model(inp)
- outPath='./yolov5/runs/detect/exp/'+inp.split('/')[2]
-  #out_path='./yolov5/runs/detect/exp/'+inp_vid.split('/')[2]
- return outPath
-
-inp=gr.inputs.Image() 
-output=gr.outputs.Image() 
+ g = (size / max(inp.size))  # gain
+ #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
+ results = model(inp)  # inference
+ results.render()  # updates results.imgs with boxes and labels
+ return Image.fromarray(results.imgs[0])
+ 
+
+inp = gr.inputs.Image(type='pil', label="Original Image")
+output = gr.outputs.Image(type="pil", label="Output Image")
 
 
 io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='Party Symbol Detection',examples=['Content/4.jpg'])