Update app.py

app.py

@@ -148,10 +148,9 @@ def predict(
         gray_image_pil = Image.fromarray(gray_image).convert('L')
     else:
         gray_image_pil = input_image["layers"][0]
-    if pipe is None:
-        base_model_path = 'black-forest-labs/FLUX.1-dev'
-        lora_path = 'theSure/Omnieraser'
-        pipe = load_model(base_model_path, lora_path)
+    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"),