Update app.py

app.py

@@ -6,10 +6,7 @@ from PIL import Image
                            
 #subprocess.run(["mv","content/custom_data.yaml","./yolov5/data"])         
 
-def find(name, path):
- for root, dirs, files in os.walk(path):
-  if name in files:
-   return os.path.join(root, name)
+
 
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
 model.conf = 0.38  
@@ -23,7 +20,7 @@ def detect(inp):
  return Image.fromarray(results.imgs[0])
  
 
-inp = gr.inputs.Image(type='pil', label="Original Image")
+inp = gr.inputs.Image(type='file', label="Original Image")
 output = gr.outputs.Image(type="pil", label="Output Image")