app.py

import torch
import os
import requests

MODEL_URL = "https://huggingface.co/datasets/uruguayai/fooocus/resolve/main/juggernautXL_v8Rundiffusion.safetensors"
MODEL_PATH = "/home/user/app/juggernautXL_v8Rundiffusion.safetensors"

def download_file(url, path):
    """Descarga un archivo desde una URL si no está presente en la ruta especificada."""
    if not os.path.exists(path):
        response = requests.get(url)
        with open(path, 'wb') as f:
            f.write(response.content)
        print(f"Downloaded {path}")

def load_model():
    download_file(MODEL_URL, MODEL_PATH)
    # 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():
    model = load_model()
    print("Model loaded successfully!")

if __name__ == "__main__":
    main()