Update README.md
Browse filesOutput format added
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': ['Nathen15@hotmail.com']}
|
| 148 |
+
|
| 149 |
```
|