Update app.py

app.py

@@ -14,9 +14,6 @@ import sys
 import tempfile
 from typing import Sequence, Mapping, Any, Union
 import asyncio
-import execution
-from nodes import init_extra_nodes
-import server
 
 # Copy functions from FluxSimpleUpscaler.txt
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
@@ -58,13 +55,13 @@ def add_extra_model_paths() -> None:
 def import_custom_nodes() -> None:
     import asyncio
     import execution
-    from nodes import init_extra_nodes
+    from nodes import init_custom_nodes
     import server
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     server_instance = server.PromptServer(loop)
     execution.PromptQueue(server_instance)
-    init_extra_nodes()
+    init_custom_nodes()
 
 # Setup ComfyUI and custom nodes
 if not os.path.exists("ComfyUI"):
@@ -282,11 +279,11 @@ def enhance_image(
 
     return [resized_input, image]
 
-with gr.Blocks(css=css, title="🎨 AI Image Upscaler - FLUX ComfyUI") as demo:
+with gr.Blocks(css=css, title="🎨 Flux Image Upscaler") as demo:
     gr.HTML("""
     <div class="main-header">
-        <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>
+        <h1>🎨 Flux Image Upscaler</h1>
+        <p>Upload an image or provide a URL to upscale it using FLUX FP8 with Ultimate SD Upscale</p>
         <p>Using FLUX.1-dev FP8 model</p>
     </div>
     """)