ezcz commited on
Commit
49add9e
1 Parent(s): 0955aae

Update space

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -5,7 +5,7 @@ from datetime import datetime
5
 
6
  class BrightLlamaChatbot:
7
  def __init__(self):
8
- self.model_id = "ezcz/bright-llama-3b-chat"
9
  self.pipe = pipeline(
10
  "text-generation",
11
  model=self.model_id,
@@ -46,7 +46,7 @@ class BrightLlamaChatbot:
46
  def create_interface(self):
47
  with gr.Blocks(css=self.get_custom_css()) as interface:
48
  gr.HTML("<h1>🦙 Bright Llama Chatbot</h1>")
49
- gr.HTML("<p>An AI assistant specialized in coding and reasoning tasks.</p>")
50
 
51
  with gr.Row():
52
  with gr.Column(scale=4):
@@ -70,12 +70,11 @@ class BrightLlamaChatbot:
70
  clear_btn = gr.Button("Clear")
71
 
72
  with gr.Column(scale=1):
73
- gr.HTML("<h3>Features</h3>")
74
  gr.HTML("""
75
  <ul>
76
  <li>💬 Natural conversation</li>
77
- <li>🎨 Code syntax highlighting</li>
78
- <li>⚡ Quick responses</li>
79
  </ul>
80
  """)
81
 
 
5
 
6
  class BrightLlamaChatbot:
7
  def __init__(self):
8
+ self.model_id = "ezcz/Llama-3.2-3B-BrightLlamaChat-LoRA"
9
  self.pipe = pipeline(
10
  "text-generation",
11
  model=self.model_id,
 
46
  def create_interface(self):
47
  with gr.Blocks(css=self.get_custom_css()) as interface:
48
  gr.HTML("<h1>🦙 Bright Llama Chatbot</h1>")
49
+ gr.HTML("<p>An AI assistant using ezcz/Llama-3.2-3B-BrightLlamaChat-LoRA</p>")
50
 
51
  with gr.Row():
52
  with gr.Column(scale=4):
 
70
  clear_btn = gr.Button("Clear")
71
 
72
  with gr.Column(scale=1):
73
+ gr.HTML("<h3>Bright Llama LoRA Chatbot:</h3>")
74
  gr.HTML("""
75
  <ul>
76
  <li>💬 Natural conversation</li>
77
+ <li>🐢Running on CPU - expect slow responses</li>
 
78
  </ul>
79
  """)
80