ab-ai commited on
Commit
68fb4eb
·
verified ·
1 Parent(s): d28552d

Update README.md

Browse files

Output format added

Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -144,5 +144,6 @@ inputs = tokenizer(model_prompt, return_tensors="pt").to(device)
144
  # adjust max_new_tokens according to your need
145
  outputs = model.generate(**inputs, do_sample=True, max_new_tokens=120)
146
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
147
- print(response)
 
148
  ```
 
144
  # adjust max_new_tokens according to your need
145
  outputs = model.generate(**inputs, do_sample=True, max_new_tokens=120)
146
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
147
+ print(response) #{'middlename': ['Abner'], 'dob': ['23/03/1926'], 'email': ['[email protected]']}
148
+
149
  ```