namnh113 commited on
Commit
6dbfedb
1 Parent(s): 24a233a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ class LLM_Langchain():
31
 
32
  model_name = return_sum_models(model_parent)
33
  list_model = [model_name]
34
- if model_parent == "python":
35
  list_model += [model_name+"_v2"]
36
  if model_parent != "cpp":
37
  list_model += ["Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"]
 
31
 
32
  model_name = return_sum_models(model_parent)
33
  list_model = [model_name]
34
+ if model_parent in ["python", "java"]:
35
  list_model += [model_name+"_v2"]
36
  if model_parent != "cpp":
37
  list_model += ["Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"]