Fix unet statedict load

app.py

@@ -28,9 +28,6 @@ os.environ["CUDA_MODULE_LOADING"] = "LAZY"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 weight_dtype = torch.float16 if device == "cuda" else torch.float32
 
-CHECKPOINT_DIR = "./checkpoints/VITONHD/model"
-os.makedirs(CHECKPOINT_DIR, exist_ok=True)
-
 def load_models():
     """Загружает все необходимые модели"""
     print("⚙️ Загрузка моделей...")
@@ -67,17 +64,11 @@ def load_models():
             subfolder="unet"
         )
         
-        unet_checkpoint_path = os.path.join(CHECKPOINT_DIR, "pytorch_model.bin")
-        if not os.path.exists(unet_checkpoint_path):
-            print("⏳ Загрузка чекпоинта модели...")
-            temp_path = hf_hub_download(
-                repo_id="Benrise/VITON-HD",
-                filename="VITONHD/model/pytorch_model.bin",
-                token=TOKEN
-            )
-            import shutil
-            shutil.copy(temp_path, unet_checkpoint_path)
-            print(f"✅ Файл успешно сохранён по пути: {unet_checkpoint_path}")
+        unet_checkpoint_path = hf_hub_download(
+            repo_id="Benrise/VITON-HD",
+            filename="VITONHD/model/pytorch_model.bin",
+            token=TOKEN
+        )
         unet.load_state_dict(torch.load(unet_checkpoint_path, map_location=device))
         
         cloth_encoder = ClothEncoder.from_pretrained(
@@ -187,7 +178,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 900px}
     )
 
 if __name__ == "__main__":
-    demo.queue(concurrency_count=1, max_size=2).launch(
+    demo.queue(max_size=1).launch(
         server_name="0.0.0.0" if os.getenv("SPACE_ID") else None,
         share=os.getenv("GRADIO_SHARE") == "True"
     )