commit

app.py

@@ -14,8 +14,11 @@ class Examples(gr.helpers.Examples):
         self.create()
 
 
+HF_TOKEN = os.environ.get('HF_KEY')
+
 client = Client("Canyu/Diception",
-                max_workers=3)
+                max_workers=3,
+                hf_token=HF_TOKEN)
 
 
 def process_image_check(path_input):
@@ -26,8 +29,8 @@ def process_image_check(path_input):
 
 def infer_image_matting(matting_image_input):
     return client.predict(
-      file=handle_file(matting_image_input),
-      api_name="/infer_image_matting"
+      prompt=handle_file(matting_image_input),
+      api_name="/test"
     )
 
 def clear_cache():