waleyWang commited on
Commit
2626977
·
verified ·
1 Parent(s): 0d0729f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -34,10 +34,10 @@ PROMPT_DICT = {
34
 
35
  class Args:
36
  def __init__(self):
37
- self.bert_rounter = "waleyWang/CO2RRChat/tree/main/bert_router_model"
38
- self.bert_model = "waleyWang/CO2RRChat/tree/main/product_predict"
39
- self.com_plan_model = "waleyWang/CO2RRChat/tree/main/computational_plan"
40
- self.code_model = "waleyWang/CO2RRChat/tree/main/code_generate"
41
  self.context_size = 32768
42
  self.max_gen_len = 30000
43
  self.cache_dir = "./cache"
 
34
 
35
  class Args:
36
  def __init__(self):
37
+ self.bert_rounter = "waleyWang/CO2RRChat/bert_router_model"
38
+ self.bert_model = "waleyWang/CO2RRChat/product_predict"
39
+ self.com_plan_model = "waleyWang/CO2RRChat/computational_plan"
40
+ self.code_model = "waleyWang/CO2RRChat/code_generate"
41
  self.context_size = 32768
42
  self.max_gen_len = 30000
43
  self.cache_dir = "./cache"