Update app.py

app.py

@@ -8,18 +8,28 @@ 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("temp.jpg")],
-      paths=None,
-      batch_size=1,
-      input_size=320,
-      output_dir='output',
-      visualization=True)
+def infer(webcam, img,option):
+  if option == "webcam":
+      webcam.save('temp.jpg')
+      result = model.Segmentation(
+          images=[cv2.imread("temp.jpg")],
+          paths=None,
+          batch_size=1,
+          input_size=320,
+          output_dir='output',
+          visualization=True)
+  else:
+      img.save('temp.jpg')
+      result = model.Segmentation(
+          images=[cv2.imread("temp.jpg")],
+          paths=None,
+          batch_size=1,
+          input_size=320,
+          output_dir='output',
+          visualization=True)
   return result[0]['front'][:,:,::-1], result[0]['mask']
 
-inputs = gr.inputs.Image(source="webcam", label="Webcam", type="pil")
+inputs = [gr.inputs.Image(source="webcam", label="Webcam", type="pil",optional=True),gr.inputs.Image(source="upload", label="Input Image", type="pil",optional=True),gr.inputs.Radio(choices=["webcam","Image"], type="value", default="Image", label="Input Type")]
 outputs = [
            gr.outputs.Image(type="numpy",label="Front"),
            gr.outputs.Image(type="numpy",label="Mask")