Update

.gitignore

@@ -0,0 +1,2 @@
+# Ignore local
+local/

app.py

@@ -27,10 +27,11 @@ MASK_CONTEXT_PADDING = 16 * 8
 
 api = HfApi()
 
+model_name = "2025-01-11_22-00-18-save-10359-55-129_patched.safetensors"
+
 # Download the blanchon/FurnitureFlags init Path(__file__).parent / examples_dataset
 FLAG_PATH = Path(__file__).parent / "examples_dataset"
 if not torch.cuda.is_available():
-    FLAG_PATH = Path(__file__).parent / "examples_dataset"
     FLAG_PATH.mkdir(parents=True, exist_ok=True)
 else:
     api.snapshot_download(
@@ -47,7 +48,7 @@ for flag_file in flag_files:
         reader = csv.reader(file)
         next(reader)
         for row in reader:
-            furniture_image, room_image, results_values, time = row
+            furniture_image, room_image, results_values, time, *others = row
             room_image = json.loads(room_image)
             room_image_background = room_image["background"]
             room_image_layers = room_image["layers"]
@@ -75,7 +76,7 @@ if not torch.cuda.is_available():
 else:
     state_dict, network_alphas = FluxFillPipeline.lora_state_dict(
         pretrained_model_name_or_path_or_dict="blanchon/FluxFillFurniture",
-        weight_name="2025-01-11_22-00-18-save-10359-55-129_patched.safetensors",
+        weight_name=model_name,
         return_alphas=True,
     )
 
@@ -226,14 +227,16 @@ def flag(
         return
     callback.flag(
         flag_data=[furniture_image_input, room_image_input, results],
+        flag_option=model_name,
     )
-    # Upload the flagged data points to the hub
-    api.upload_folder(
-        repo_id="blanchon/FurnitureFlags",
-        repo_type="dataset",
-        folder_path=FLAG_PATH,
-        ignore_patterns=[".cache"],
-    )
+    if torch.cuda.is_available():
+        # Upload the flagged data points to the hub
+        api.upload_folder(
+            repo_id="blanchon/FurnitureFlags",
+            repo_type="dataset",
+            folder_path=FLAG_PATH,
+            ignore_patterns=[".cache"],
+        )
 
 
 @spaces.GPU(duration=150)
@@ -446,6 +449,7 @@ with gr.Blocks(css=css) as demo:
             )
             furniture_examples = gr.Examples(
                 examples=list({example[0] for example in EXAMPLES.values()}),
+                label="Furniture examples",
                 examples_per_page=6,
                 inputs=[furniture_image_input],
             )
@@ -471,6 +475,7 @@ with gr.Blocks(css=css) as demo:
             )
             room_examples = gr.Examples(
                 examples=[example[1] for example in EXAMPLES.values()],
+                label="Room examples",
                 examples_per_page=6,
                 # examples=[
                 #     make_example(
@@ -583,7 +588,7 @@ with gr.Blocks(css=css) as demo:
         show_case = gr.Examples(
             examples=list(EXAMPLES.values()),
             inputs=[furniture_image_input, room_image_input, results],
-            label=None,
+            label="Examples",
             examples_per_page=12,
         )
 

uv.lock

@@ -240,11 +240,11 @@ requires-dist = [
 
 [[package]]
 name = "fsspec"
-version = "2024.12.0"
+version = "2024.9.0"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/ee/11/de70dee31455c546fbc88301971ec03c328f3d1138cfba14263f651e9551/fsspec-2024.12.0.tar.gz", hash = "sha256:670700c977ed2fb51e0d9f9253177ed20cbde4a3e5c0283cc5385b5870c8533f", size = 291600 }
+sdist = { url = "https://files.pythonhosted.org/packages/62/7c/12b0943011daaaa9c35c2a2e22e5eb929ac90002f08f1259d69aedad84de/fsspec-2024.9.0.tar.gz", hash = "sha256:4b0afb90c2f21832df142f292649035d80b421f60a9e1c027802e5a0da2b04e8", size = 286206 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/de/86/5486b0188d08aa643e127774a99bac51ffa6cf343e3deb0583956dca5b22/fsspec-2024.12.0-py3-none-any.whl", hash = "sha256:b520aed47ad9804237ff878b504267a3b0b441e97508bd6d2d8774e3db85cee2", size = 183862 },
+    { url = "https://files.pythonhosted.org/packages/1d/a0/6aaea0c2fbea2f89bfd5db25fb1e3481896a423002ebe4e55288907a97a3/fsspec-2024.9.0-py3-none-any.whl", hash = "sha256:a0947d552d8a6efa72cc2c730b12c41d043509156966cca4fb157b0f2a0c574b", size = 179253 },
 ]
 
 [[package]]