codegenerator / app.py
Rahul Kumar
this is initial
d8152da
raw
history blame contribute delete
504 Bytes
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
model_id = "deepseek-ai/DeepSeek-Coder-6.7B-Instruct"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto")
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
prompt = "Add a custom field 'fabric_type' to product admin page in OpenCart and save it to database."
result = pipe(prompt, max_new_tokens=400)
print(result[0]['generated_text'])