Update app.py
Browse files
app.py
CHANGED
@@ -34,8 +34,8 @@ def predict(message, max_new_tokens=4056, temperature=3.5, top_p=0.9, top_k=800)
|
|
34 |
do_sample=True
|
35 |
)
|
36 |
response = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
|
37 |
-
return response
|
38 |
-
|
39 |
DESCRIPTION = """
|
40 |
# 👋🏻Welcome to 🙋🏻♂️Tonic's🧑🏻🚀YI-200K🚀"
|
41 |
You can use this Space to test out the current model [Tonic/YI](https://huggingface.co/01-ai/Yi-34B)
|
|
|
34 |
do_sample=True
|
35 |
)
|
36 |
response = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
|
37 |
+
return [("bot", response)]
|
38 |
+
|
39 |
DESCRIPTION = """
|
40 |
# 👋🏻Welcome to 🙋🏻♂️Tonic's🧑🏻🚀YI-200K🚀"
|
41 |
You can use this Space to test out the current model [Tonic/YI](https://huggingface.co/01-ai/Yi-34B)
|