TDN-M commited on
Commit
8521f37
·
verified ·
1 Parent(s): 4a1f84e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=2048)
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: #f0f0f0;}" # Thêm CSS tùy chỉnh
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