Update app.py

app.py

@@ -55,13 +55,13 @@ def add_extra_model_paths() -> None:
 def import_custom_nodes() -> None:
     import asyncio
     import execution
-    from nodes import init_custom_nodes
+    from nodes import init_extra_nodes
     import server
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     server_instance = server.PromptServer(loop)
     execution.PromptQueue(server_instance)
-    init_custom_nodes()
+    init_extra_nodes()
 
 # Setup ComfyUI and custom nodes
 if not os.path.exists("ComfyUI"):
@@ -92,7 +92,7 @@ if not os.path.exists(t5_path):
 
 vae_path = "ComfyUI/models/vae/ae.safetensors"
 if not os.path.exists(vae_path):
-    hf_hub_download("black-forest-labs/FLUX.1-dev", "ae.safetensors", subfolder="vae", local_dir="ComfyUI/models/vae")
+    hf_hub_download("black-forest-labs/FLUX.1-dev", "ae.safetensors", local_dir="ComfyUI/models/vae")
 
 esrgan_x2_path = "ComfyUI/models/upscale_models/RealESRGAN_x2.pth"
 if not os.path.exists(esrgan_x2_path):
@@ -279,11 +279,11 @@ def enhance_image(
 
     return [resized_input, image]
 
-with gr.Blocks(css=css, title="🎨 Flux Image Upscaler") as demo:
+with gr.Blocks(css=css, title="🎨 AI Image Upscaler - FLUX ComfyUI") as demo:
     gr.HTML("""
     <div class="main-header">
-        <h1>🎨 Flux Image Upscaler</h1>
-        <p>Upload an image or provide a URL to upscale it using FLUX FP8 with Ultimate SD Upscale</p>
+        <h1>🎨 AI Image Upscaler (ComfyUI Workflow)</h1>
+        <p>Upload an image or provide a URL to upscale it using FLUX FP8 with ComfyUI Ultimate SD Upscale</p>
         <p>Using FLUX.1-dev FP8 model</p>
     </div>
     """)