Update app.py

app.py

@@ -45,7 +45,7 @@ def download_file(repo_id, filename, local_dir):
                 local_files_only=False
             )
             logger.info(f"Downloaded to {downloaded_path}")
-            return downloaded_path  # Return the actual path from hf_hub_download
+            return downloaded_path
         except Exception as e:
             logger.error(f"Download failed: {e}")
             raise
@@ -85,10 +85,10 @@ except Exception as e:
     logger.error(f"Failed to load SDXL base model: {e}")
     raise
 
-# Load Kolors unet weights
+# Load Kolors unet weights with weights_only=False
 logger.info(f"Loading Kolors unet weights from {kolors_weights}...")
 try:
-    state_dict = torch.load(kolors_weights, map_location=device)
+    state_dict = torch.load(kolors_weights, map_location=device, weights_only=False)
     pipe.unet.load_state_dict(state_dict)
     logger.info("Kolors unet weights loaded successfully.")
 except Exception as e: