import cohere
import gradio as gr
import os
from dotenv import load_dotenv
load_dotenv()
COHERE_API_KEY = os.getenv('COHERE_API_KEY')
# OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
co = cohere.Client(os.environ["COHERE_API_KEY"])

response = co.generate(
    prompt='Please briefly explain to me how Deep Learning works on hardware devices using at most 100 words.',
    max_tokens=200
)
print(response.generations[0].text)