Spaces:

Baptlem
/

UCDR-Net

Runtime error

Baptlem commited on May 2, 2023

Commit

7b66f42

•

1 Parent(s): 71f4cfe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -91,7 +91,7 @@ def pipe_inference(
     prompt_ids = shard(prompt_ids)
     negative_prompt_ids = shard(negative_prompt_ids)
     processed_image = shard(processed_image)
     output = pipe(
         prompt_ids=prompt_ids,
         image=processed_image,

     prompt_ids = shard(prompt_ids)
     negative_prompt_ids = shard(negative_prompt_ids)
     processed_image = shard(processed_image)
+    print("Inference...")
     output = pipe(
         prompt_ids=prompt_ids,
         image=processed_image,