Update app.py

app.py

@@ -1,13 +1,13 @@
+import gradio as gr
 from safetensors.torch import load_file
 import os
 import requests
 
-# URLs de los archivos
+# URLs y rutas de archivos
 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)
         if response.status_code == 200:
@@ -18,15 +18,16 @@ def download_file(url, path):
             raise Exception(f"Failed to download {url}, status code: {response.status_code}")
 
 def load_model():
-    # Descargar el archivo si no existe
     download_file(MODEL_URL, MODEL_PATH)
-    # Cargar los pesos utilizando safetensors
-    model = load_file(MODEL_PATH)  # Carga el modelo desde el archivo safetensors
+    model = load_file(MODEL_PATH)  # Cargar modelo con safetensors
     return model
 
-def main():
-    model = load_model()
-    print("Model loaded successfully with safetensors!")
+def predict(input_text):
+    # Función de predicción simple
+    return f"Model loaded and ready: {input_text}"
+
+iface = gr.Interface(fn=predict, inputs="text", outputs="text")
 
 if __name__ == "__main__":
-    main()
+    load_model()  # Asegúrate de que el modelo esté cargado
+    iface.launch(server_name="0.0.0.0", server_port=7860)