one more test with user HF_TOKEN

app.py

@@ -125,8 +125,40 @@ def process_mask(
     return mask
 
 
+def set_client_for_session(request: gr.Request):
+    x_ip_token = request.headers['x-ip-token']
+    return Client("SkalskiP/florence-sam-masking", headers={"X-IP-Token": x_ip_token})
+
+
 @spaces.GPU(duration=100)
+def run_flux(
+    image: Image.Image,
+    mask: Image.Image,
+    prompt: str,
+    seed_slicer: int,
+    randomize_seed_checkbox: bool,
+    strength_slider: float,
+    num_inference_steps_slider: int,
+    resolution_wh: Tuple[int, int],
+) -> Image.Image:
+    width, height = resolution_wh
+    if randomize_seed_checkbox:
+        seed_slicer = random.randint(0, MAX_SEED)
+    generator = torch.Generator().manual_seed(seed_slicer)
+    return PIPE(
+        prompt=prompt,
+        image=image,
+        mask_image=mask,
+        width=width,
+        height=height,
+        strength=strength_slider,
+        generator=generator,
+        num_inference_steps=num_inference_steps_slider
+    ).images[0]
+
+
 def process(
+    client,
     input_image_editor: dict,
     inpainting_prompt_text: str,
     masking_prompt_text: str,
@@ -135,8 +167,7 @@ def process(
     seed_slicer: int,
     randomize_seed_checkbox: bool,
     strength_slider: float,
-    num_inference_steps_slider: int,
-    progress=gr.Progress(track_tqdm=True)
+    num_inference_steps_slider: int
 ):
     if not inpainting_prompt_text:
         gr.Info("Please enter a text prompt.")
@@ -171,25 +202,21 @@ def process(
     image = image.resize((width, height), Image.LANCZOS)
     mask = mask.resize((width, height), Image.LANCZOS)
     mask = process_mask(mask, mask_inflation=mask_inflation_slider, mask_blur=mask_blur_slider)
-
-    if randomize_seed_checkbox:
-        seed_slicer = random.randint(0, MAX_SEED)
-    generator = torch.Generator().manual_seed(seed_slicer)
-    result = PIPE(
-        prompt=inpainting_prompt_text,
+    image = run_flux(
         image=image,
-        mask_image=mask,
-        width=width,
-        height=height,
-        strength=strength_slider,
-        generator=generator,
-        num_inference_steps=num_inference_steps_slider
-    ).images[0]
-    print('INFERENCE DONE')
-    return result, mask
+        mask=mask,
+        prompt=inpainting_prompt_text,
+        seed_slicer=seed_slicer,
+        randomize_seed_checkbox=randomize_seed_checkbox,
+        strength_slider=strength_slider,
+        num_inference_steps_slider=num_inference_steps_slider,
+        resolution_wh=(width, height)
+    )
+    return image, mask
 
 
 with gr.Blocks() as demo:
+    client_component = gr.State()
     gr.Markdown(MARKDOWN)
     with gr.Row():
         with gr.Column():
@@ -304,6 +331,7 @@ with gr.Blocks() as demo:
     submit_button_component.click(
         fn=process,
         inputs=[
+            client_component,
             input_image_editor_component,
             inpainting_prompt_text_component,
             masking_prompt_text_component,
@@ -319,5 +347,6 @@ with gr.Blocks() as demo:
             output_mask_component
         ]
     )
+    demo.load(set_client_for_session, None, client_component)
 
 demo.launch(debug=False, show_error=True)