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': ['[email protected]']}
|
148 |
+
|
149 |
```
|