Biochemistry3.0 / app.py
ziyadsuper2017's picture
Update app.py
7c64638
raw
history blame contribute delete
702 Bytes
import gradio as gr
import cohere
# Initialize the Cohere client
co = cohere.Client('mTeiGiDIpi4R7perkfRnMnCmi5jJzM8QziDVPrG8') # Replace with your api_key
def generate_response(prompt, temperature):
output = co.generate(
model='9b2e329d-7542-4c44-8f8c-cac03a6c4f5a-ft',
prompt=prompt,
temperature=0.4 # adjust the temperature here
)
response = output.generations[0].text
return response
iface = gr.Interface(fn=generate_response,
inputs="text",
outputs="text",
title='Cohere Text Generation',
description='Enter your prompt to generate text')
iface.launch(share=True)