Update app.py

app.py

@@ -22,10 +22,11 @@ class LLM_Langchain():
         else:
             model_name_visibility = False
             
-        self.model_name = return_sum_models(model_parent)
-        checkpoint = st.sidebar.selectbox(
-            label = "Model used",
-            options = [self.model_name],
+        model_name = return_sum_models(model_parent)
+        
+        self.checkpoint = st.sidebar.selectbox(
+            label = "Choose Model (nam194/... is my model)",
+            options = [model_name, "Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"],
             help="Model used to predict",
             disabled=model_name_visibility
             )
@@ -81,7 +82,7 @@ class LLM_Langchain():
         
         
         llm = HuggingFaceHub(
-            repo_id = self.model_name,
+            repo_id = self.checkpoint,
             model_kwargs = self.model_kwargs
         )