Update app.py

app.py

@@ -38,18 +38,27 @@ class LLM_Langchain():
             value=120,
             help="Set the max tokens to get accurate results"
             )
+
+        self.num_beams = st.sidebar.slider(
+            label="num beams",
+            min_value=1,
+            max_value=10,
+            step=1,
+            value=4,
+            help="Set num beam"
+            )
         
         self.top_k = st.sidebar.slider(
-            label="top_k",
-            min_value=1.0,
-            max_value=50.0,
-            step=1.0,
-            value=30.0,
+            label="top k",
+            min_value=1,
+            max_value=50,
+            step=1,
+            value=30,
             help="Set the top_k"
             )
         
         self.top_p = st.sidebar.slider(
-            label="top_p",
+            label="top p",
             min_value=0.1,
             max_value=1.0,
             step=0.05,
@@ -61,7 +70,8 @@ class LLM_Langchain():
         self.model_kwargs = {
             "max_new_tokens": self.max_new_tokens,
             "top_k": self.top_k,
-            "top_p": self.top_p
+            "top_p": self.top_p,
+            "num_beams": self.num_beams
         }
 
         os.environ['HUGGINGFACEHUB_API_TOKEN'] = self.API_KEY