Update app.py

app.py

@@ -14,9 +14,8 @@ HF_TOKEN_UPSCALER = os.environ.get("HF_TOKEN_UPSCALER")
 client = AsyncInferenceClient()
 llm_client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# Directorio de almacenamiento de imágenes
 DATA_PATH = Path("./data")
-DATA_PATH.mkdir(exist_ok=True)  # Asegura que el directorio exista
+DATA_PATH.mkdir(exist_ok=True)  
 
 def enable_lora(lora_add, basemodel):
     return basemodel if not lora_add else lora_add
@@ -99,9 +98,7 @@ def get_storage():
     usage = sum([f['size'] for f in files])
     return [file["name"] for file in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
 
-css = """
-#col-container{ margin: 0 auto; max-width: 1024px;}
-"""
+css = """#col-container{ margin: 0 auto; max-width: 1024px;}"""
 
 with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
     with gr.Column(elem_id="col-container"):
@@ -141,8 +138,9 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
                 )
         with gr.Row():
             with gr.Column():
-                file_list = gr.Gallery(label="Imágenes Guardadas")  # Usar Gallery en lugar de Files
+                file_list = gr.Gallery(label="Imágenes Guardadas")  
                 storage_info = gr.Text(label="Uso de Almacenamiento")
-            refresh_btn = gr.Button("Actualizar Galería")
-            refresh_btn.click(fn=get_storage, inputs=None, outputs=[file_list, storage_info])
+                refresh_btn = gr.Button("Actualizar Galería")
+                refresh_btn.click(fn=get_storage, inputs=None, outputs=[file_list, storage_info])
+                
     demo.launch(allowed_paths=[str(DATA_PATH)])