Update app.py

app.py

@@ -2,17 +2,8 @@ import torch
 import os
 import requests
 
-# Definir las URLs de los archivos en el dataset de Hugging Face
 MODEL_URL = "https://huggingface.co/datasets/uruguayai/fooocus/resolve/main/juggernautXL_v8Rundiffusion.safetensors"
-INTERPOSER_URL = "https://huggingface.co/datasets/uruguayai/fooocus/resolve/main/xl-to-v1_interposer-v4.0.safetensors"
-VAE_URL = "https://huggingface.co/datasets/uruguayai/fooocus/resolve/main/vaeapp_sd15.pt"
-EXPANSION_URL = "https://huggingface.co/datasets/uruguayai/fooocus/resolve/main/fooocus_expansion.bin"
-
-# Definir las rutas locales donde se guardarán los archivos
 MODEL_PATH = "/home/user/app/juggernautXL_v8Rundiffusion.safetensors"
-INTERPOSER_PATH = "/home/user/app/xl-to-v1_interposer-v4.0.safetensors"
-VAE_PATH = "/home/user/app/vaeapp_sd15.pt"
-EXPANSION_PATH = "/home/user/app/fooocus_expansion.bin"
 
 def download_file(url, path):
     """Descarga un archivo desde una URL si no está presente en la ruta especificada."""
@@ -23,14 +14,9 @@ def download_file(url, path):
         print(f"Downloaded {path}")
 
 def load_model():
-    # Descargar los archivos necesarios
     download_file(MODEL_URL, MODEL_PATH)
-    download_file(INTERPOSER_URL, INTERPOSER_PATH)
-    download_file(VAE_URL, VAE_PATH)
-    download_file(EXPANSION_URL, EXPANSION_PATH)
-
-    # Cargar el modelo; asegúrate de usar weights_only=True por razones de seguridad
-    model = torch.load(MODEL_PATH, weights_only=True)
+    # Cargar con weights_only=False si confías en la fuente del modelo
+    model = torch.load(MODEL_PATH, weights_only=False)
     return model
 
 def main():