winglian commited on
Commit
a62ccaa
1 Parent(s): e49e33e

add updated model endpoints

Browse files
Files changed (1) hide show
  1. app.py +6 -9
app.py CHANGED
@@ -80,8 +80,11 @@ class Pipeline:
80
 
81
 
82
  AVAILABLE_MODELS = {
83
- "hermes-13b": "l96d2cswh431qj",
84
- "manticore-13b": "jifr1oczbrmr3n",
 
 
 
85
  }
86
 
87
  _memoized_models = defaultdict()
@@ -98,12 +101,6 @@ start_message = """- The Assistant is helpful and transparent.
98
  """
99
 
100
 
101
- def generate_responses(system_prompt, user_query, model_choice):
102
- # Prepare the prompt
103
- prompt = system_prompt + " " + user_query
104
-
105
-
106
-
107
  def user(message, nudge_msg, history1, history2):
108
  history1 = history1 or []
109
  history2 = history2 or []
@@ -215,7 +212,7 @@ with gr.Blocks() as arena:
215
  with gr.Row():
216
  with gr.Column():
217
  rlhf_persona = gr.Textbox(
218
- "", label="Persona Tags", interactive=True, visible=True, placeholder="Tell us about how you are judging the quality. ex: #SFW #NSFW #helpful #ethical #creativity", lines=1)
219
  message = gr.Textbox(
220
  label="What do you want to ask?",
221
  placeholder="Ask me anything.",
 
80
 
81
 
82
  AVAILABLE_MODELS = {
83
+ "hermes-13b": "p0zqb2gkcwp0ww",
84
+ "manticore-13b-chat": "u6tv84bpomhfei",
85
+ "airoboros-13b": "rglzxnk80660ja",
86
+ "supercot-13b": "0be7865dwxpwqk",
87
+ "mpt-7b-instruct": "jpqbvnyluj18b0",
88
  }
89
 
90
  _memoized_models = defaultdict()
 
101
  """
102
 
103
 
 
 
 
 
 
 
104
  def user(message, nudge_msg, history1, history2):
105
  history1 = history1 or []
106
  history2 = history2 or []
 
212
  with gr.Row():
213
  with gr.Column():
214
  rlhf_persona = gr.Textbox(
215
+ "", label="Persona Tags", interactive=True, visible=True, placeholder="Tell us about how you are judging the quality. ex: #CoT #SFW #NSFW #helpful #ethical #creativity", lines=1)
216
  message = gr.Textbox(
217
  label="What do you want to ask?",
218
  placeholder="Ask me anything.",