Update app.py

app.py

@@ -95,8 +95,14 @@ MODEL_PATH = {
 }
 
 
-def inference(img_path, model, focus_face=None):
-    print(img_path, model, focus_face)
+def inference(upload, webcam, model, focus_face=None):
+    print(upload, webcam, model, focus_face)
+    if upload is not None:
+        img_path = upload
+    elif upload is None and webcam is not None:
+        img_path = webcam
+    else:
+        img_path = ""
     mat, scale = load_image(img_path, focus_face)
     output = convert(mat, model, scale)
     save_path = f"output/out.{img_path.rsplit('.')[-1]}"
@@ -109,8 +115,9 @@ title = "AnimeGANv2: To produce your own animation 😶‍🌫️"
 description = r"""
 ### 🔥Demo AnimeGANv2: To produce your own animation. <br>
 #### How to use:
-1. Upload your image
-2. Select the style (*For human*: PortraitSketch, JP_face; *For scene*: Hayao, Shinkai, Paprika)
+1a. Upload your image
+1b. Use webcam to take an image
+2. Select the style (**For human**: PortraitSketch, JP_face; **For scene**: Hayao, Shinkai, Paprika)
 3. Choice of whether to extract the face.(Warning: Yes if there is a face in the image)
 """
 article = r"""
@@ -125,7 +132,8 @@ examples=[['sample1.jpg', 'PortraitSketch', "Yes"],
           ['sample6.jpeg', 'JP_face', "No"]]
 gr.Interface(
     inference, [
-        gr.Image(type="filepath", label="Input image"),
+        gr.Image(type="filepath", label="Image", source="upload"),
+        gr.Image(type="filepath", label="Webcam", source="webcam"),
         gr.Dropdown([
             'Hayao',
             'Shinkai',