budikomarudin commited on
Commit
c3752f4
·
verified ·
1 Parent(s): 36de1b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -7
app.py CHANGED
@@ -7,18 +7,12 @@ llm = ChatOpenAI(
7
  base_url="https://api.featherless.ai/v1/",
8
  api_key=api_key,
9
  streaming=True,
10
- # model="Qwen/Qwen3-32B",
11
- model="deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
12
- # model="meta-llama/Llama-3.3-70B-Instruct",
13
- # model="mistralai/Magistral-Small-2506",
14
- # model="unsloth/DeepSeek-R1-Distill-Llama-70B",
15
- # model="unsloth/Qwen2.5-72B-Instruct",
16
- # model="unsloth/Llama-3.3-70B-Instruct",
17
  )
18
 
19
  def respond(
20
  message,
21
  history: list[tuple[str, str]],
 
22
  system_message,
23
  max_tokens,
24
  temperature,
@@ -33,6 +27,7 @@ def respond(
33
  messages.append({"role": "assistant", "content": val[1]})
34
 
35
  messages.append({"role": "user", "content": message})
 
36
  llm.max_tokens = max_tokens
37
  llm.temperature = temperature
38
  llm.top_p = top_p
@@ -46,6 +41,19 @@ def respond(
46
  demo = gr.ChatInterface(
47
  respond,
48
  additional_inputs=[
 
 
 
 
 
 
 
 
 
 
 
 
 
49
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
50
  gr.Slider(minimum=1, maximum=16384, value=2048, step=1, label="Max new tokens"),
51
  gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
 
7
  base_url="https://api.featherless.ai/v1/",
8
  api_key=api_key,
9
  streaming=True,
 
 
 
 
 
 
 
10
  )
11
 
12
  def respond(
13
  message,
14
  history: list[tuple[str, str]],
15
+ model,
16
  system_message,
17
  max_tokens,
18
  temperature,
 
27
  messages.append({"role": "assistant", "content": val[1]})
28
 
29
  messages.append({"role": "user", "content": message})
30
+ llm.model = model
31
  llm.max_tokens = max_tokens
32
  llm.temperature = temperature
33
  llm.top_p = top_p
 
41
  demo = gr.ChatInterface(
42
  respond,
43
  additional_inputs=[
44
+ gr.Dropdown(
45
+ [
46
+ "Qwen/Qwen3-32B",
47
+ "Qwen/Qwen2.5-72B-Instruct",
48
+ "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
49
+ "meta-llama/Llama-3.3-70B-Instruct",
50
+ "mistralai/Magistral-Small-2506",
51
+ "unsloth/DeepSeek-R1-Distill-Llama-70B",
52
+ "unsloth/Qwen2.5-72B-Instruct",
53
+ "unsloth/Llama-3.3-70B-Instruct",
54
+ ],
55
+ label="Models"
56
+ ),
57
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
58
  gr.Slider(minimum=1, maximum=16384, value=2048, step=1, label="Max new tokens"),
59
  gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),