slower fix

app.py

@@ -30,7 +30,6 @@ pipe = FluxFillPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16
 ).to("cuda")
 
-simple_lama = SimpleLama()
 
 def prepare_image_and_mask(
     image,
@@ -149,11 +148,14 @@ def mask_generation(image=None, d=None):
 
     return out
 
-def erase(image=None,mask=None) : 
+
+def erase(image=None, mask=None):
+    simple_lama = SimpleLama()
     image = load_img(image)
     mask = load_img(mask).convert("L")
     return simple_lama(image, mask)
 
+
 @spaces.GPU
 def main(*args):
     api_num = args[0]
@@ -166,7 +168,7 @@ def main(*args):
         return inpaint(*args)
     elif api_num == 4:
         return mask_generation(*args)
-    elif api_num == 5 : 
+    elif api_num == 5:
         return erase(*args)