Update app.py
Browse files
app.py
CHANGED
@@ -21,6 +21,8 @@ summary.append("")
|
|
21 |
main_point.append("")
|
22 |
|
23 |
models=[
|
|
|
|
|
24 |
"google/gemma-7b",
|
25 |
"google/gemma-7b-it",
|
26 |
"google/gemma-2b",
|
@@ -29,8 +31,6 @@ models=[
|
|
29 |
"codellama/CodeLlama-70b-Instruct-hf",
|
30 |
"openchat/openchat-3.5-0106",
|
31 |
"NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
|
32 |
-
"mistralai/Mixtral-8x7B-Instruct-v0.1",
|
33 |
-
"mistralai/Mixtral-8x7B-Instruct-v0.2",
|
34 |
]
|
35 |
|
36 |
client_z=[]
|
@@ -105,6 +105,7 @@ def compress_history(formatted_prompt):
|
|
105 |
|
106 |
def comment_generate(prompt, history,post_check, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.0,):
|
107 |
#def question_generate(prompt, history):
|
|
|
108 |
full_conv=history
|
109 |
print("###############\nRUNNING QUESTION GENERATOR\n###############\n")
|
110 |
seed = random.randint(1,1111111111111111)
|
@@ -137,8 +138,8 @@ def comment_generate(prompt, history,post_check, agent_name=agents[0], sys_promp
|
|
137 |
|
138 |
post_check['comment']=post_check['comment']+1
|
139 |
#out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
|
140 |
-
full_conv.append((
|
141 |
-
html_out=load_html(full_conv,
|
142 |
|
143 |
#history.append((output,history))
|
144 |
#[textbox, chatbot, textbox, json, json, html]
|
|
|
21 |
main_point.append("")
|
22 |
|
23 |
models=[
|
24 |
+
"mistralai/Mixtral-8x7B-Instruct-v0.1",
|
25 |
+
"mistralai/Mixtral-8x7B-Instruct-v0.2",
|
26 |
"google/gemma-7b",
|
27 |
"google/gemma-7b-it",
|
28 |
"google/gemma-2b",
|
|
|
31 |
"codellama/CodeLlama-70b-Instruct-hf",
|
32 |
"openchat/openchat-3.5-0106",
|
33 |
"NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
|
|
|
|
|
34 |
]
|
35 |
|
36 |
client_z=[]
|
|
|
105 |
|
106 |
def comment_generate(prompt, history,post_check, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1028, top_p=0.95, repetition_penalty=1.0,):
|
107 |
#def question_generate(prompt, history):
|
108 |
+
print(post_check)
|
109 |
full_conv=history
|
110 |
print("###############\nRUNNING QUESTION GENERATOR\n###############\n")
|
111 |
seed = random.randint(1,1111111111111111)
|
|
|
138 |
|
139 |
post_check['comment']=post_check['comment']+1
|
140 |
#out_json = {'user':"",'datetime':current_time,'title':title,'blog':1,'comment':0,'reply':0,"prompt":prompt,"output":output}
|
141 |
+
full_conv.append((output,))
|
142 |
+
html_out=load_html(full_conv,None)
|
143 |
|
144 |
#history.append((output,history))
|
145 |
#[textbox, chatbot, textbox, json, json, html]
|