Update app.py

app.py

@@ -7,21 +7,26 @@ from PIL import Image
 #subprocess.run(["mv","content/custom_data.yaml","./yolov5/data"])         
 
 
+def load_model():
+ '''
+ Loading hub model & setting the preferences for the model  
+ '''
+ model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
+ model.conf = 0.38 
+ model.hide_conf=True
+ return model
 
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
-model.conf = 0.38  
-model.hide_conf=True
 #, force_reload=True
 def detect(inp):
  #g = (size / max(inp.size))  # gain
  #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
- inp=Image.open(inp)
+ model=load_model()
  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='file', label="Original Image")
+inp = gr.inputs.Image(type="pil", label="Original Image")
 output = gr.outputs.Image(type="pil", label="Output Image")