Omnibus commited on
Commit
95ce061
·
verified ·
1 Parent(s): 49eebbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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((history,))
141
- html_out=load_html(full_conv,title)
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]