Spaces:
Runtime error
Runtime error
add updated model endpoints
Browse files
app.py
CHANGED
@@ -80,8 +80,11 @@ class Pipeline:
|
|
80 |
|
81 |
|
82 |
AVAILABLE_MODELS = {
|
83 |
-
"hermes-13b": "
|
84 |
-
"manticore-13b": "
|
|
|
|
|
|
|
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.",
|