Spaces:

wencheng256
/

DiffusionRAWSpace

Sleeping

wencheng256 commited on Apr 3

Commit

5407ff1

1 Parent(s): 6176e59

update

Files changed (2) hide show

__pycache__/pipeline_controlnet_sd_xl_raw.cpython-310.pyc ADDED Viewed

Binary file (54.8 kB). View file

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ def infer_fn(prompt, mask_edited, raw_tensor_state, mask_tensor_state, cond_tens
             control_image=cond_t
         ).images[0]
-        return tensor_to_pil(result), mask_edited
     except Exception as e:
         traceback.print_exc()
@@ -115,7 +115,7 @@ def build_demo():
                 mask_tensor_state,
                 cond_tensor_state
             ],
-            outputs=[output_image, rgb_display]
         )
     return demo

             control_image=cond_t
         ).images[0]
+        return tensor_to_pil(result)
     except Exception as e:
         traceback.print_exc()
                 mask_tensor_state,
                 cond_tensor_state
             ],
+            outputs=[output_image]
         )
     return demo