Revert "test running API calls with user HF_TOKEN"

This reverts commit 24aa6b97f1fb8ce9581cd9e40402fd163fc896d3.

app.py

@@ -1,9 +1,10 @@
-import random
 from typing import Tuple
 
-import gradio as gr
-import numpy as np
+import os
 import requests
+import random
+import numpy as np
+import gradio as gr
 import spaces
 import torch
 from PIL import Image, ImageFilter
@@ -22,8 +23,8 @@ MAX_SEED = np.iinfo(np.int32).max
 IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
-# HF_TOKEN = os.environ.get("HF_TOKEN", None)
-# client = Client("SkalskiP/florence-sam-masking", hf_token=HF_TOKEN)
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+client = Client("SkalskiP/florence-sam-masking", hf_token=HF_TOKEN)
 
 
 def remove_background(image: Image.Image, threshold: int = 50) -> Image.Image:
@@ -43,7 +44,6 @@ def remove_background(image: Image.Image, threshold: int = 50) -> Image.Image:
 
 EXAMPLES = [
     [
-        None,
         {
             "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-image.png", stream=True).raw),
             "layers": [remove_background(Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-mask-2.png", stream=True).raw))],
@@ -57,7 +57,6 @@ EXAMPLES = [
         30
     ],
     [
-        None,
         {
             "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-image.png", stream=True).raw),
             "layers": [remove_background(Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-mask-3.png", stream=True).raw))],
@@ -102,14 +101,8 @@ def is_image_empty(image: Image.Image) -> bool:
     return all(pixel == 0 for pixel in pixels)
 
 
-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 process(
-    client,
     input_image_editor: dict,
     inpainting_prompt_text: str,
     masking_prompt_text: str,
@@ -171,7 +164,6 @@ def process(
 
 
 with gr.Blocks() as demo:
-    client_component = gr.State()
     gr.Markdown(MARKDOWN)
     with gr.Row():
         with gr.Column():
@@ -248,7 +240,6 @@ with gr.Blocks() as demo:
                 input_image_editor_component,
                 inpainting_prompt_text_component,
                 masking_prompt_text_component,
-                masking_prompt_text_component,
                 seed_slicer_component,
                 randomize_seed_checkbox_component,
                 strength_slider_component,
@@ -278,6 +269,5 @@ with gr.Blocks() as demo:
             output_mask_component
         ]
     )
-    demo.load(set_client_for_session, None, client_component)
 
 demo.launch(debug=False, show_error=True)