Weyaxi commited on
Commit
1d7a1fb
1 Parent(s): 1c7ae94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -67,12 +67,12 @@ PLACEHOLDER = f"""
67
  def respond(
68
  message,
69
  history: list[tuple[str, str]],
 
70
  max_tokens,
71
  temperature,
72
  top_p,
73
  top_k,
74
  repeat_penalty,
75
- model,
76
  ):
77
  from llama_cpp import Llama
78
  from llama_cpp_agent import LlamaCppAgent
@@ -130,6 +130,7 @@ def respond(
130
  demo = gr.ChatInterface(
131
  respond,
132
  additional_inputs=[
 
133
  gr.Slider(minimum=1, maximum=8192, value=8192, step=1, label="Max tokens"),
134
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
135
  gr.Slider(
@@ -153,8 +154,6 @@ demo = gr.ChatInterface(
153
  step=0.1,
154
  label="Repetition penalty",
155
  ),
156
-
157
- gr.Dropdown(files, value=files[0], label="Model"),
158
  ],
159
  theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue", neutral_hue="gray",font=[gr.themes.GoogleFont("Exo"), "ui-sans-serif", "system-ui", "sans-serif"]).set(
160
  body_background_fill_dark="#0f172a",
@@ -172,7 +171,7 @@ demo = gr.ChatInterface(
172
  undo_btn="Undo",
173
  clear_btn="Clear",
174
  submit_btn="Send",
175
- description="Chat Einstein 🧑‍🔬 Einstein v6.1 Llama3-8B",
176
  chatbot=gr.Chatbot(scale=1, placeholder=PLACEHOLDER)
177
  )
178
 
 
67
  def respond(
68
  message,
69
  history: list[tuple[str, str]],
70
+ model,
71
  max_tokens,
72
  temperature,
73
  top_p,
74
  top_k,
75
  repeat_penalty,
 
76
  ):
77
  from llama_cpp import Llama
78
  from llama_cpp_agent import LlamaCppAgent
 
130
  demo = gr.ChatInterface(
131
  respond,
132
  additional_inputs=[
133
+ gr.Dropdown(files, value=files[0], label="Model"),
134
  gr.Slider(minimum=1, maximum=8192, value=8192, step=1, label="Max tokens"),
135
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
136
  gr.Slider(
 
154
  step=0.1,
155
  label="Repetition penalty",
156
  ),
 
 
157
  ],
158
  theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue", neutral_hue="gray",font=[gr.themes.GoogleFont("Exo"), "ui-sans-serif", "system-ui", "sans-serif"]).set(
159
  body_background_fill_dark="#0f172a",
 
171
  undo_btn="Undo",
172
  clear_btn="Clear",
173
  submit_btn="Send",
174
+ description="# Chat Einstein 🧑‍🔬 Einstein v6.1 Llama3-8B\n\nSelect your model in the additional input place",
175
  chatbot=gr.Chatbot(scale=1, placeholder=PLACEHOLDER)
176
  )
177