Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
|
4 |
# Tải mô hình đã upload từ Hugging Face
|
5 |
model_name = "TDN-M/tknv1" # Thay thế bằng tên người dùng và tên repository của bạn
|
6 |
-
pipe = pipeline("text-generation", model=model_name, max_new_tokens=
|
7 |
|
8 |
def chatbot(message, history):
|
9 |
# Tạo tin nhắn từ người dùng
|
@@ -21,7 +21,7 @@ iface = gr.ChatInterface(
|
|
21 |
title="Chatbot sử dụng tiếng Việt",
|
22 |
description="Một chatbot sử dụng mô hình đã được huấn luyện bằng tiếng Việt.",
|
23 |
theme="default", # Chọn theme cho giao diện
|
24 |
-
css=".gradio-container {background-color: #
|
25 |
)
|
26 |
|
27 |
# Chạy ứng dụng
|
|
|
3 |
|
4 |
# Tải mô hình đã upload từ Hugging Face
|
5 |
model_name = "TDN-M/tknv1" # Thay thế bằng tên người dùng và tên repository của bạn
|
6 |
+
pipe = pipeline("text-generation", model=model_name, max_new_tokens=512)
|
7 |
|
8 |
def chatbot(message, history):
|
9 |
# Tạo tin nhắn từ người dùng
|
|
|
21 |
title="Chatbot sử dụng tiếng Việt",
|
22 |
description="Một chatbot sử dụng mô hình đã được huấn luyện bằng tiếng Việt.",
|
23 |
theme="default", # Chọn theme cho giao diện
|
24 |
+
css=".gradio-container {background-color: #fef0f0;}" # Thêm CSS tùy chỉnh
|
25 |
)
|
26 |
|
27 |
# Chạy ứng dụng
|