Upload folder using huggingface_hub

app.py

@@ -22,7 +22,7 @@ pipe_t2i = MageFlowPipeline.from_pretrained(T2I_MODEL, device="cuda")
 pipe_edit = MageFlowPipeline.from_pretrained(EDIT_MODEL, device="cuda")
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=60)
 def generate(
     prompt: str,
     negative_prompt: str,
@@ -58,7 +58,7 @@ def generate(
     return img
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=60)
 def edit(
     prompt: str,
     negative_prompt: str,
@@ -106,7 +106,7 @@ CSS = """
 .dark .gradio-container { color: var(--body-text-color); }
 """
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
+with gr.Blocks(css=CSS) as demo:
     gr.Markdown(
         "# Mage-Flow\n"
         "Efficient Native-Resolution Foundation Model for Image Generation and Editing.\n\n"
@@ -206,4 +206,4 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
         )
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(theme=gr.themes.Citrus(), mcp_server=True, show_error=True)