Update README.md
Browse files
README.md
CHANGED
@@ -97,7 +97,7 @@ messages = [
|
|
97 |
{"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
|
98 |
{"role": "user", "content": "Who are you?"},
|
99 |
]
|
100 |
-
chatbot = pipeline("text-generation", model="behbudiy/Mistral-7B-Instruct-Uz")
|
101 |
chatbot(messages)
|
102 |
```
|
103 |
|
|
|
97 |
{"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
|
98 |
{"role": "user", "content": "Who are you?"},
|
99 |
]
|
100 |
+
chatbot = pipeline("text-generation", model="behbudiy/Mistral-7B-Instruct-Uz", device='cuda')
|
101 |
chatbot(messages)
|
102 |
```
|
103 |
|