nam194 commited on
Commit
70849f0
1 Parent(s): e002a1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- list_model += ["Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"]
 
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)",