Spaces:
Runtime error
Runtime error
File size: 2,788 Bytes
9a06ebe f66ca1f 9a06ebe f66ca1f 9a06ebe 1700c40 9a06ebe 1700c40 9a06ebe f66ca1f 1700c40 9a06ebe 39495a2 9a06ebe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import logging
import os
import time
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=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):
for _ in range(num_lines):
st.write("")
st.set_page_config(layout="centered", page_icon="🍔", page_title="Recipe Generation Using Cohere")
# Check if COHERE_API_KEY is not set from secrets.toml or os.environ
if "CO_API_KEY" not in os.environ:
raise KeyError("CO_API_KEY not found in st.secrets or os.environ. Please set it in "
".streamlit/secrets.toml or as an environment variable.")
# Adding a header to direct users to sign up for Cohere, explore the playground,
# and check out our git repo.
st.header("Recipe Generation")
#st.selectbox("Choose a cuisine:", options=['mexican', 'italian'], key="cuisine")
st.write("What would you like a recipe for?")
st.session_state.disabled = False
with st.form("form"):
dish = st.text_input("Dish", placeholder=DEFAULT_DISH)
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(label="Generate Recipe!")
if submitted:
with st.spinner("Cooking up something delicious ... please wait a few seconds ... "):
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:
prompt += f" Make sure it is {', '.join(options)}."
logger.info(f"Sending prompt {repr(prompt)}")
start = time.time()
gens = command(co, prompt)
recipe = gens[0].text
num_tokens = len(gens[0].token_likelihoods)
logger.info(f"Generated recipe with {num_tokens} tokens in {time.time()-start:.1f}s: {repr(recipe)}")
st.markdown(f"## Recipe for {' '.join(options)} {dish}")
st.write(recipe)
st.markdown("## Delicious! 10/10 with rice")
# Draw chat history.
with st.expander("About", expanded=False):
st.markdown(
"This demo uses the Cohere API. Cohere provides access to advanced Large Language Models and NLP tools through one easy-to-use API. "
"[**Get started for free!**]"
"(https://dashboard.cohere.ai/welcome/register?utm_source=cohere-owned&utm_"
"medium=content&utm_campaign=sandbox&utm_term=streamlit&utm_content=conversant)")
|