amornpan commited on
Commit
93fb29f
1 Parent(s): 70f9199

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -80,8 +80,8 @@ model = AutoModelForCausalLM.from_pretrained("amornpan/openthaigpt-MedChatModelv
80
  input_text = "ใส่คำถามทางการแพทย์ที่นี่"
81
  inputs = tokenizer(input_text, return_tensors="pt")
82
 
83
- # Generate the output
84
- output = model.generate(**inputs)
85
 
86
  # Decode and print the generated response, skipping special tokens
87
- print(tokenizer.decode(output[0], skip_special_tokens=True))
 
80
  input_text = "ใส่คำถามทางการแพทย์ที่นี่"
81
  inputs = tokenizer(input_text, return_tensors="pt")
82
 
83
+ # Generate the output with a higher max length or max new tokens
84
+ output = model.generate(**inputs, max_new_tokens=100, do_sample=True, temperature=0.7)
85
 
86
  # Decode and print the generated response, skipping special tokens
87
+ print(tokenizer.decode(output[0], skip_special_tokens=True))