Update app.py

app.py

@@ -7,8 +7,6 @@ import gradio as gr
 from huggingface_hub import HfApi, hf_hub_download
 from safetensors.torch import load_file, save_file
 
-# --- ARCHITECTURE PROFILES ---
-# Defines which sensitive layers must stay in BF16 during INT8 quantization to prevent precision collapse.
 ARCH_PROFILES = {
     "FLUX / Generic Rectified Flow": ["norm", "ln_", "embed", "time_in", "vector_in", "guidance_in", "txt_in", "img_in"],
     "Z-Image / DiT Core": ["t_embedder", "cap_embedder", "all_x_embedder", "all_final_layer", "rope_embedder", "embed_tokens", "norm", "ln_", "shared"],
@@ -122,7 +120,8 @@ def toggle_int8_warning(precision):
     return gr.update(visible=(precision == "INT8"))
 
 # --- GUI ---
-with gr.Blocks(theme=gr.themes.Base(primary_hue="blue", neutral_hue="slate")) as demo:
+# FIXED: Removed the theme argument from gr.Blocks()
+with gr.Blocks() as demo:
     gr.Markdown(
         """
         # ⚡ Universal Model Quantizer Hub
@@ -132,7 +131,6 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="blue", neutral_hue="slate")) as
     )
 
     with gr.Row():
-        # LEFT COLUMN: Configuration
         with gr.Column(scale=5):
             with gr.Tabs():
                 
@@ -183,28 +181,22 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="blue", neutral_hue="slate")) as
             
             start_btn = gr.Button("🚀 Start Cloud Quantization", variant="primary", size="lg")
 
-        # RIGHT COLUMN: Logs
         with gr.Column(scale=4):
             output_log = gr.Textbox(
                 label="Terminal Output", 
                 lines=24, 
                 interactive=False, 
-                max_lines=30,
-                show_copy_button=True
+                max_lines=30
             )
 
-    # --- WIRING ---
-    # Presets
     preset_flux.click(lambda: ("black-forest-labs/FLUX.2-klein-9B", "FLUX / Generic Rectified Flow"), outputs=[source_repo, arch_profile])
     preset_zimage.click(lambda: ("your-username/Z-Image-Turbo", "Z-Image / DiT Core"), outputs=[source_repo, arch_profile])
     preset_sd3.click(lambda: ("stabilityai/stable-diffusion-3.5-large", "Stable Diffusion (SDXL/SD3)"), outputs=[source_repo, arch_profile])
 
-    # Dynamic Updates
     source_repo.change(fn=generate_target_repo, inputs=[source_repo, precision], outputs=[target_repo])
     precision.change(fn=generate_target_repo, inputs=[source_repo, precision], outputs=[target_repo])
     precision.change(fn=toggle_int8_warning, inputs=[precision], outputs=[int8_warning])
 
-    # Execution
     start_btn.click(
         fn=convert_and_upload, 
         inputs=[hf_token, source_repo, target_repo, precision, target_components, arch_profile], 
@@ -212,4 +204,4 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="blue", neutral_hue="slate")) as
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Base(primary_hue="blue", neutral_hue="slate"))