Update to v3 model

app.py

@@ -222,9 +222,9 @@ def get_model():
         
         print("Loading SD-Gcode model...")
         
-        # Download config and weights
-        config_path = hf_hub_download("twarner/dcode-sd-gcode", "config.json")
-        weights_path = hf_hub_download("twarner/dcode-sd-gcode", "pytorch_model.bin")
+        # Download config and weights from v3 model
+        config_path = hf_hub_download("twarner/dcode-sd-gcode-v3", "config.json")
+        weights_path = hf_hub_download("twarner/dcode-sd-gcode-v3", "pytorch_model.bin")
         
         with open(config_path) as f:
             config = json.load(f)
@@ -302,8 +302,8 @@ def get_model():
         
         # Load gcode tokenizer
         try:
-            # Try loading custom tokenizer
-            tokenizer_path = hf_hub_download("twarner/dcode-sd-gcode", "gcode_tokenizer/tokenizer.json")
+            # Try loading custom tokenizer from v3 model
+            tokenizer_path = hf_hub_download("twarner/dcode-sd-gcode-v3", "gcode_tokenizer/tokenizer.json")
             gcode_tokenizer = PreTrainedTokenizerFast(tokenizer_file=tokenizer_path)
             print("Loaded custom gcode tokenizer")
         except Exception:
@@ -675,7 +675,7 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
         gcode_output = gr.Code(label=None, language=None, lines=12)
     
     gr.Markdown("---")
-    gr.Markdown("machine: 841×1189mm / pen servo 40-90° / [github](https://github.com/Twarner491/dcode) / [model](https://huggingface.co/twarner/dcode-sd-gcode) / mit")
+    gr.Markdown("machine: 841×1189mm / pen servo 40-90° / [github](https://github.com/Twarner491/dcode) / [model](https://huggingface.co/twarner/dcode-sd-gcode-v3) / mit")
     
     generate_btn.click(generate, [prompt, temperature, max_tokens, num_steps, guidance], [gcode_output, preview])
     prompt.submit(generate, [prompt, temperature, max_tokens, num_steps, guidance], [gcode_output, preview])

requirements.txt

@@ -3,5 +3,7 @@ gradio_client==1.3.0
 torch
 diffusers
 transformers
+tokenizers
 accelerate
 huggingface_hub
+spaces