Spaces:

vinoku89
/

svg-code-generator

Running on Zero

vinoku89 commited on Jul 26, 2025

Commit

0702272

verified ·

1 Parent(s): 42f22af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import torch
 import spaces
 # Load model
-tokenizer = AutoTokenizer.from_pretrained("vinoku89/svg-code-generator")
-model = AutoModelForCausalLM.from_pretrained("vinoku89/svg-code-generator")
 # Move model to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import spaces
 # Load model
+tokenizer = AutoTokenizer.from_pretrained("vinoku89/qwen3-4B-svg-code-gen")
+model = AutoModelForCausalLM.from_pretrained("vinoku89/qwen3-4B-svg-code-gen")
 # Move model to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")