tokens

app.py

@@ -6,17 +6,17 @@ import cohere
 import streamlit as st
 
 DEFAULT_DISH = "herring and banana quesadillas"
+MAX_TOKENS = 300
 
 logger = logging.getLogger('recipe-logger')
 
 
-def command(co, prompt, model='command-xlarge-20221108', max_tokens=500, *args, **kwargs):
+def command(co, prompt, model='command-xlarge-20221108', max_tokens=MAX_TOKENS, *args, **kwargs):
     gens = co.generate(prompt, model=model, max_tokens=max_tokens, return_likelihoods='GENERATION', *args, **kwargs)
     return sorted(gens, key=lambda g: -g.likelihood)
 
 
 def space(num_lines=1):
-    """Adds empty lines to the Streamlit app."""
     for _ in range(num_lines):
         st.write("")
 
@@ -38,12 +38,14 @@ st.write("What would you like a recipe for?")
 
 with st.form("form"):
     dish = st.text_input("Dish", placeholder=DEFAULT_DISH)
-    options = st.multiselect("Special requests",
-                             ["extra delicious", "vegan", "gluten-free", "low-calorie", "low-fat", "low-sugar"])
+    options = st.multiselect("Special requests", [
+        "extra delicious", "vegan", "gluten-free", "sugar-free", "dairy-free", "nut-free", "low-calorie", "low-fat",
+        "low-sugar"
+    ])
     submitted = st.form_submit_button("Generate Recipe!")
 
 if submitted:
-    dish = dish or DEFAULT_DISH
+    dish = (dish or DEFAULT_DISH).strip()
     co = cohere.Client(api_key=os.environ['CO_API_KEY'])
     prompt = f"Give a recipe for {dish}."
     if options:

requirements.txt

@@ -0,0 +1 @@
+cohere==2.8.0