Views
No views yet
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline1def get_pipline():
2 model_name = "nirusanan/Qwen3-ReactJs-code"
3
4 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
5 tokenizer.pad_token = tokenizer.eos_token
6
7 model = AutoModelForCausalLM.from_pretrained(
8 model_name,
9 torch_dtype=torch.float16,
10 device_map="cuda:0",
11 trust_remote_code=True
12 )
13
14 pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=3500)
15
16 return pipe
17
18pipe = get_pipline()1def generate_prompt(project_title, description):
2 prompt = f"""Below is an instruction that describes a project. Write Reactjs code to accomplish the project described below.
3
4### Instruction:
5Project:
6{project_title}
7
8Project Description:
9{description}
10
11### Response:
12"""
13 return prompt1prompt = generate_prompt(project_title = "Your ReactJs project", description = "Your ReactJs project description")
2result = pipe(prompt)
3generated_text = result[0]['generated_text']
4print(generated_text.split("### End")[0])