K00B404 commited on
Commit
72c3d12
·
verified ·
1 Parent(s): c49e8f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -31,9 +31,9 @@ def chat_with_model(user_input, system_prompt, selected_model):
31
  result = client.chat.completions.create(
32
  model=selected_model,
33
  messages=messages,
34
- temperature=0.5,
35
- max_tokens=2048,
36
- top_p=0.7,
37
  stream=False # Stream disabled for simplicity
38
  )
39
  return result["choices"][0]["message"]["content"]
@@ -52,8 +52,8 @@ def get_prompt(name):
52
 
53
  # List of available models
54
  available_models = [
55
- "gpt-3.5-turbo",
56
- "gpt-4",
57
  "HuggingFaceH4/zephyr-7b-beta",
58
  "HuggingFaceH4/zephyr-7b-alpha"
59
  ]
 
31
  result = client.chat.completions.create(
32
  model=selected_model,
33
  messages=messages,
34
+ temperature=0.9,
35
+ max_tokens=512,
36
+ top_p=0.97,
37
  stream=False # Stream disabled for simplicity
38
  )
39
  return result["choices"][0]["message"]["content"]
 
52
 
53
  # List of available models
54
  available_models = [
55
+ "Qwen/Qwen2.5-Coder-0.5B-Instruct",
56
+ "Qwen/Qwen2.5-Coder-1.5B-Instruct",
57
  "HuggingFaceH4/zephyr-7b-beta",
58
  "HuggingFaceH4/zephyr-7b-alpha"
59
  ]