Update README.md
Browse files
README.md
CHANGED
@@ -74,13 +74,13 @@ pline = pipeline(
|
|
74 |
|
75 |
# Example of response generation
|
76 |
message = [{"role": "user", "content": "Kateri je najpomembnejši dogodek v slovenski zgodovini?"}]
|
77 |
-
response =
|
78 |
print("Model's response:", response[0]["generated_text"][-1]["content"])
|
79 |
|
80 |
# Example of conversation chain
|
81 |
new_message = response[0]["generated_text"]
|
82 |
new_message.append({"role": "user", "content": "Lahko bolj podrobno opišeš ta dogodek?"})
|
83 |
-
response =
|
84 |
print("Model's response:", response[0]["generated_text"][-1]["content"])
|
85 |
```
|
86 |
|
|
|
74 |
|
75 |
# Example of response generation
|
76 |
message = [{"role": "user", "content": "Kateri je najpomembnejši dogodek v slovenski zgodovini?"}]
|
77 |
+
response = pline(message, max_new_tokens=512)
|
78 |
print("Model's response:", response[0]["generated_text"][-1]["content"])
|
79 |
|
80 |
# Example of conversation chain
|
81 |
new_message = response[0]["generated_text"]
|
82 |
new_message.append({"role": "user", "content": "Lahko bolj podrobno opišeš ta dogodek?"})
|
83 |
+
response = pline(new_message, max_new_tokens=1024)
|
84 |
print("Model's response:", response[0]["generated_text"][-1]["content"])
|
85 |
```
|
86 |
|