Spaces:

theSure
/

Omnieraser

Running on Zero

theSure commited on Apr 7

Commit

b597317

verified ·

1 Parent(s): 50a6090

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,6 @@ def predict(
         gray_image_pil = input_image["layers"][0]
     base_model_path = 'black-forest-labs/FLUX.1-dev'
     lora_path = 'theSure/Omnieraser'
-    pipe = load_model(base_model_path, lora_path)
     result = pipe(
         prompt=prompt,
         control_image=input_image["background"].convert("RGB"),

         gray_image_pil = input_image["layers"][0]
     base_model_path = 'black-forest-labs/FLUX.1-dev'
     lora_path = 'theSure/Omnieraser'
     result = pipe(
         prompt=prompt,
         control_image=input_image["background"].convert("RGB"),