Upload app.py

app.py

@@ -8,15 +8,22 @@ from PIL import Image, ImageDraw, ImageFilter
 from diffusers import StableDiffusionXLInpaintPipeline
 from insightface.app import FaceAnalysis
 import insightface
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 
 # ==============================================================================
 # 1. SETUP DELL'AMBIENTE E DOWNLOAD MODELLI
 # ==============================================================================
 
+# Download sicuro e robusto di antelopev2 da Hugging Face per evitare errori di unzip
+print("Download modelli AntelopeV2 da Hugging Face...")
+try:
+    snapshot_download("DIAMONIK7777/antelopev2", local_dir="./models/antelopev2")
+except Exception as e:
+    print(f"Errore download AntelopeV2: {e}")
+
 # Inizializza l'analizzatore di volti (InsightFace)
 print("Inizializzazione FaceAnalysis...")
-face_app = FaceAnalysis(name='antelopev2', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
+face_app = FaceAnalysis(name='antelopev2', root='./', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
 face_app.prepare(ctx_id=0, det_size=(640, 640))
 
 # Download e caricamento del modello Inswapper