Update app.py

app.py

@@ -18,23 +18,23 @@ class LLM_Langchain():
             help="Type in your HuggingFace API key to use this app")
 
             
-        model_parent = st.sidebar.selectbox(
+        self.model_parent = st.sidebar.selectbox(
             label = "Choose language",
             options = ["python", "java", "javascript", "php", "ruby", "go", "cpp"],
             help="Choose languages",
             )
 
-        if model_parent is None:
+        if self.model_parent is None:
             model_name_visibility = True
         else:
             model_name_visibility = False
             
-        model_name = return_sum_models(model_parent)
+        model_name = return_sum_models(self.model_parent)
         list_model = [model_name]
-        if model_parent in ["python", "java"]:
+        if self.model_parent in ["python", "java"]:
             list_model += [model_name+"_v2"]
-        if model_parent != "cpp":
-            list_model += ["Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"]
+        if self.model_parent != "cpp":
+            list_model += ["Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{self.model_parent}"]
         
         self.checkpoint = st.sidebar.selectbox(
             label = "Choose model (namnh113/... is my model)",
@@ -92,7 +92,7 @@ class LLM_Langchain():
 
     def generate_response(self, input_text):
         
-        
+        input_text = "Summarize " + self.model_parent.capitalize() + ": " + input_text
         llm = HuggingFaceHub(
             repo_id = self.checkpoint,
             model_kwargs = self.model_kwargs