commit

app.py

@@ -4,6 +4,8 @@ from gradio_client import Client, handle_file
 from pathlib import Path
 from gradio.utils import get_cache_folder
 
+import torch
+
 from PIL import Image
 
 class Examples(gr.helpers.Examples):
@@ -63,6 +65,7 @@ def process_image_4(image_path, prompt):
 
     inputs = []
     for p in prompt:
+        cur_p = map_prompt[p]
         image = Image.open(image_path)
 
         w, h = image.size
@@ -78,7 +81,7 @@ def process_image_4(image_path, prompt):
                 'input_images': image.unsqueeze(0),
                 'original_size': torch.tensor([[w,h]]),
                 'target_size': torch.tensor([[768, 768]]),
-                'prompt': [p],
+                'prompt': [cur_p],
                 'coor_point': coor_point,
                 'point_labels': point_labels,
                 'generator': generator
@@ -88,11 +91,13 @@ def process_image_4(image_path, prompt):
     return inputs
 
 
-def infer_image_matting(image_path, prompt):
+def inf(image_path, prompt):
+    print(image_path)
+    print(prompt)
     inputs = process_image_4(image_path, prompt)
-    return None
+    # return None
     return client.predict(
-      batch=inputs,
+      data=inputs,
       api_name="/inf"
     )
 
@@ -154,9 +159,10 @@ def run_demo_server():
             queue=False,
         ).success(
             # fn=process_pipe_matting,
-            fn=infer_image_matting,
+            fn=inf,
             inputs=[
                 matting_image_input,
+                checkbox_group
             ],
             outputs=[matting_image_output],
             concurrency_limit=1,
@@ -176,11 +182,11 @@ def run_demo_server():
         )
 
         gr.Examples(
-            fn=infer_image_matting,
+            fn=inf,
             examples=[
-                "assets/person.jpg",              
+                ["assets/person.jpg", ['depth', 'normal', 'entity', 'pose']]             
             ],
-            inputs=[matting_image_input],
+            inputs=[matting_image_input, checkbox_group],
             outputs=[matting_image_output],
             cache_examples=True,
             # cache_examples=False,

requirements.txt

@@ -0,0 +1,7 @@
+accelerate
+diffusers
+invisible_watermark
+torch
+transformers
+xformers
+sentencepiece