yasserrmd commited on
Commit
4e12807
·
verified ·
1 Parent(s): 7986cd2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -25,7 +25,7 @@ This model was trained using AutoTrain. For more information, please visit [Auto
25
 
26
  from transformers import AutoModelForCausalLM, AutoTokenizer
27
 
28
- model_path = "PATH_TO_THIS_REPO"
29
 
30
  tokenizer = AutoTokenizer.from_pretrained(model_path)
31
  model = AutoModelForCausalLM.from_pretrained(
@@ -36,7 +36,7 @@ model = AutoModelForCausalLM.from_pretrained(
36
 
37
  # Prompt content: "hi"
38
  messages = [
39
- {"role": "user", "content": "hi"}
40
  ]
41
 
42
  input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
 
25
 
26
  from transformers import AutoModelForCausalLM, AutoTokenizer
27
 
28
+ model_path = "yasserrmd/qwen2.5-html-0.5b"
29
 
30
  tokenizer = AutoTokenizer.from_pretrained(model_path)
31
  model = AutoModelForCausalLM.from_pretrained(
 
36
 
37
  # Prompt content: "hi"
38
  messages = [
39
+ {"role": "user", "content": "generate a sample html for dashboard"}
40
  ]
41
 
42
  input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')