Fix config

server.py

@@ -47,7 +47,6 @@ original_keywords = st.multiselect(
     ["parmesan cheese", "fresh oregano", "basil", "whole wheat flour"]
 )
 
-
 # st.write("Add custom ingredients here:")
 # custom_keywords = st_tags(
 #     label="",
@@ -57,8 +56,8 @@ original_keywords = st.multiselect(
 #     maxtags=15,
 #     key='1')
 
-def custom_keywords_on_change():
-    pass
+# def custom_keywords_on_change():
+#     pass
 
 
 custom_keywords = st.text_input(
@@ -79,12 +78,18 @@ submit = st.button('Get Recipe!')
 if submit:
     with st.spinner('Generating recipe...'):
         if sampling_mode == "Beam Search":
-            generated = generator(all_ingredients, return_tensors=True, return_text=False,
-                                  **beam_search.generate_kwargs)
+            generated = generator(
+                all_ingredients,
+                return_tensors=True,
+                return_text=False,
+                **beam_search.generate_kwargs)
             outputs = beam_search.post_generator(generated, tokenizer)
-        elif sampling_mode == "Top-k Sampling":
-            generated = generator(all_ingredients, return_tensors=True, return_text=False,
-                                  **top_sampling.generate_kwargs)
+        elif sampling_mode == "Top Sampling":
+            generated = generator(
+                all_ingredients,
+                return_tensors=True,
+                return_text=False,
+                **top_sampling.generate_kwargs)
             outputs = top_sampling.post_generator(generated, tokenizer)
     output = outputs[0]
     output['title'] = " ".join([w.capitalize() for w in output['title'].split()])