Update app.py
Browse files
app.py
CHANGED
@@ -34,10 +34,10 @@ PROMPT_DICT = {
|
|
34 |
|
35 |
class Args:
|
36 |
def __init__(self):
|
37 |
-
self.bert_rounter = "waleyWang/CO2RRChat/
|
38 |
-
self.bert_model = "waleyWang/CO2RRChat/
|
39 |
-
self.com_plan_model = "waleyWang/CO2RRChat/
|
40 |
-
self.code_model = "waleyWang/CO2RRChat/
|
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"
|