Views
No views yet
1# Load base model and tokenizer
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("vinoku89/svg-code-generator")
5model = AutoModelForCausalLM.from_pretrained("vinoku89/svg-code-generator")
6
7
8# Generate SVG code
9prompt = "Create a blue circle with radius 50"
10inputs = tokenizer(prompt, return_tensors="pt")
11
12# Generate with parameters
13outputs = model.generate(
14 **inputs,
15 max_length=200,
16 temperature=0.7,
17 do_sample=True,
18 pad_token_id=tokenizer.eos_token_id
19)
20
21# Decode the generated SVG code
22generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
23svg_code = generated_text[len(prompt):].strip()
24
25print("Generated SVG:")
26print(svg_code)