Omaratef3221
commited on
Commit
•
f8d4413
1
Parent(s):
de11b31
Update README.md
Browse files
README.md
CHANGED
@@ -55,4 +55,30 @@ Dialogue:
|
|
55 |
input_ids = tokenizer(prompt, return_tensors='pt').input_ids
|
56 |
output = model.generate(input_ids, top_p = 0.6, do_sample=True, temperature = 1.2, max_length = 512)
|
57 |
print(tokenizer.decode(output[0], skip_special_tokens=True).replace('#Person2#:', '\n#Person2#:').replace('#Person1#:', '\n#Person1#:'))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
```
|
|
|
55 |
input_ids = tokenizer(prompt, return_tensors='pt').input_ids
|
56 |
output = model.generate(input_ids, top_p = 0.6, do_sample=True, temperature = 1.2, max_length = 512)
|
57 |
print(tokenizer.decode(output[0], skip_special_tokens=True).replace('#Person2#:', '\n#Person2#:').replace('#Person1#:', '\n#Person1#:'))
|
58 |
+
```
|
59 |
+
#### Outputs:
|
60 |
+
```python
|
61 |
+
#Person1#: Oh, that's a nice street market. I'm glad I got to see it.
|
62 |
+
#Person2#: Yes, it is. I like the food here.
|
63 |
+
#Person1#: And the prices are reasonable.
|
64 |
+
#Person2#: I have been to this street market before, and I like it very much.
|
65 |
+
#Person1#: I'm impressed. I really like it.
|
66 |
+
#Person2#: I'm familiar with the cuisine. It is one of the best in the world.
|
67 |
+
#Person1#: What kind of food do you like?
|
68 |
+
#Person2#: I like Italian food, but I like Thai food.
|
69 |
+
#Person1#: Oh, that's really exciting. I like it very much. I think I'll take it.
|
70 |
+
#Person2#: I like Thai food too.
|
71 |
+
#Person1#: What's your favorite food?
|
72 |
+
#Person2#: I'm familiar with Thai food. I love spicy food, but I don't like spicy food.
|
73 |
+
#Person1#: Is that true?
|
74 |
+
#Person2#: Yes, that's right.
|
75 |
+
#Person1#: I like spicy food too. How about Thai food?
|
76 |
+
#Person2#: That's a great idea. I think it's really good.
|
77 |
+
#Person1#: Oh, that's a good idea. What's your favorite restaurant?
|
78 |
+
#Person2#: Oh, I'm sure I can recommend it.
|
79 |
+
#Person1#: I like Thai food, too. What do you recommend?
|
80 |
+
#Person2#: It's a very popular restaurant in China.
|
81 |
+
#Person1#: Oh, that's great.
|
82 |
+
#Person2#: I would like to try it.
|
83 |
+
#Person1#: Thanks. I'll try it.
|
84 |
```
|