Update app.py

app.py

@@ -3,28 +3,25 @@ import spaces
 import torch
 from model import ModelHandler
 from generator import Generator
-# --- IMPORT CONFIG ---
 from config import Config
 
-# 1. Initialize Models Globally (in RAM)
-# ZeroGPU will move them to VRAM inside the @spaces.GPU function
+# 1. Initialize Models Globally
 print("Initializing Application...")
 handler = ModelHandler()
 handler.load_models()
 gen = Generator(handler)
 
 # 2. Define GPU-enabled Inference Function
-@spaces.GPU(duration=20) # <-- MODIFIED
+@spaces.GPU(duration=20)
 def process_img(
     image, 
     prompt,
     negative_prompt,
-    cfg_scale, 
+    # cfg_scale,  <-- REMOVED (TCD uses 0.0)
     steps, 
     img_strength, 
     depth_strength, 
     edge_strength,
-    # tile_strength,     # <-- REMOVED
     seed
 ):
     if image is None:
@@ -32,17 +29,15 @@ def process_img(
     
     try:
         print("--- Starting Generation ---")
-        # Pass all parameters to the generator
         result = gen.predict(
             image, 
             prompt,
             negative_prompt=negative_prompt,
-            guidance_scale=cfg_scale,
+            guidance_scale=Config.CGF_SCALE, # Hardcoded to 0.0 from Config
             num_inference_steps=steps,
             img2img_strength=img_strength,
             depth_strength=depth_strength,
             lineart_strength=edge_strength,
-            # tile_strength=tile_strength,      # <-- REMOVED
             seed=seed
         )
         print("--- Generation Complete ---")
@@ -53,11 +48,11 @@ def process_img(
         raise gr.Error(f"An error occurred: {str(e)}")
 
 # 3. Build Gradio Interface
-with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Face To Style", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
-        # 🎮 Face to Pixel Art
-        Upload any image. If there is a face, we'll keep the identity. If not, we'll pixelate the scene!
+        # 🎮 Face to Style
+        Upload any image. If there is a face, we'll keep the identity. If not, we'll stylize the scene!
         """
     )
     
@@ -67,13 +62,13 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
             prompt = gr.Textbox(
                 label="Prompt (Optional)", 
                 placeholder="Leave empty for auto-captioning...",
-                info="The trigger words 'p1x3l4rt, pixel art' are added automatically."
+                info=f"The trigger words '{Config.STYLE_TRIGGER}' are added automatically."
             )
             
             negative_prompt = gr.Textbox(
                 label="Negative Prompt (Optional)", 
                 placeholder="e.g., blurry, text, watermark, bad art...",
-                value=Config.DEFAULT_NEGATIVE_PROMPT # <-- MODIFIED
+                value=Config.DEFAULT_NEGATIVE_PROMPT
             )
             
             with gr.Accordion("Advanced Settings", open=False):
@@ -84,20 +79,15 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                     precision=0
                 )
                 
-                cfg_scale = gr.Slider(
-                    elem_id="cfg_scale",
-                    minimum=1.0, 
-                    maximum=5.0, 
-                    step=0.1, 
-                    value=Config.CGF_SCALE, 
-                    label="CFG Scale"
-                )
+                # --- REMOVED CFG/GUIDANCE SLIDER ---
+                # TCD works best with guidance_scale=0.0
+                
                 steps = gr.Slider(
                     elem_id="steps",
                     minimum=4, 
                     maximum=20, 
                     step=1, 
-                    value=Config.STEPS_NUMBER, 
+                    value=8, # TCD default
                     label="Steps Number"
                 )
                 img_strength = gr.Slider(
@@ -124,25 +114,22 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                     value=Config.EDGE_STRENGTH, 
                     label="EdgeMap Strength (LineArt)"
                 )
-                # --- MODIFIED: Renamed slider ---
-                # tile_strength = gr.Slider(...) # <-- REMOVED
             
-            run_btn = gr.Button("Generate Pixel Art", variant="primary")
+            run_btn = gr.Button("Generate", variant="primary")
             
         with gr.Column(scale=1):
-            output_img = gr.Image(label="Pixel Art Result")
+            output_img = gr.Image(label="Styled Result")
                 
     # Event Handler
     all_inputs = [
         input_img, 
         prompt, 
         negative_prompt,
-        cfg_scale, 
+        # cfg_scale, <-- REMOVED
         steps, 
         img_strength, 
         depth_strength, 
         edge_strength,
-        # tile_strength,     # <-- REMOVED
         seed
     ]
     
@@ -159,5 +146,5 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_api=True # share=True is not needed on Spaces
+        show_api=True 
     )