Debug state dict keys, fix button variant

app.py

@@ -162,6 +162,11 @@ def get_model():
         print("Loading finetuned weights...")
         state_dict = torch.load(weights_path, map_location=device, weights_only=False)
         
+        # Debug: print all key prefixes
+        prefixes = set(k.split(".")[0] for k in state_dict.keys())
+        print(f"State dict prefixes: {prefixes}")
+        print(f"Sample keys: {list(state_dict.keys())[:5]}")
+        
         # Load text encoder weights
         text_encoder_state = {k.replace("text_encoder.", ""): v for k, v in state_dict.items() 
                              if k.startswith("text_encoder.")}
@@ -182,6 +187,9 @@ def get_model():
         if decoder_state:
             gcode_decoder.load_state_dict(decoder_state, strict=False)
             print(f"Loaded {len(decoder_state)} decoder weights")
+        else:
+            print("WARNING: No gcode_decoder weights found!")
+            print(f"Looking for keys starting with 'gcode_decoder.', but found: {[k for k in state_dict.keys() if 'decoder' in k.lower()][:10]}")
         
         gcode_decoder.eval()
         
@@ -458,7 +466,7 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
                 num_steps = gr.Slider(10, 50, value=20, step=5, label="diffusion steps")
                 guidance = gr.Slider(1.0, 15.0, value=7.5, step=0.5, label="guidance")
             
-            generate_btn = gr.Button("generate", variant="primary")
+            generate_btn = gr.Button("generate", variant="secondary")
             
             gr.Examples(
                 examples=[