Update app.py

app.py

@@ -9,8 +9,9 @@ torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/10/21/14/46/f
 model = hub.Module(name='U2Net')
 
 def infer(img):
+  img.save('temp.jpg')
   result = model.Segmentation(
-      images=[cv2.imread(img.name)],
+      images=[cv2.imread("temp.jpg")],
       paths=None,
       batch_size=1,
       input_size=320,
@@ -18,7 +19,7 @@ def infer(img):
       visualization=True)
   return result[0]['front'][:,:,::-1], result[0]['mask']
 
-inputs = gr.inputs.Image(type='file', label="Original Image")
+inputs = gr.inputs.Image(source="webcam", label="Webcam")
 outputs = [
            gr.outputs.Image(type="numpy",label="Front"),
            gr.outputs.Image(type="numpy",label="Mask")