Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,8 @@ class LLM_Langchain():
|
|
33 |
list_model = [model_name]
|
34 |
if model_parent == "python":
|
35 |
list_model += [model_name+"_v2"]
|
36 |
-
|
|
|
37 |
|
38 |
self.checkpoint = st.sidebar.selectbox(
|
39 |
label = "Choose model (nam194/... is my model)",
|
|
|
33 |
list_model = [model_name]
|
34 |
if model_parent == "python":
|
35 |
list_model += [model_name+"_v2"]
|
36 |
+
if model_parent != "C++":
|
37 |
+
list_model += ["Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"]
|
38 |
|
39 |
self.checkpoint = st.sidebar.selectbox(
|
40 |
label = "Choose model (nam194/... is my model)",
|