Spaces:
Sleeping
Sleeping
| 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) |