import gradio as gr
import transformers
from transformers import pipeline

def generate_text(prompt, temperature):
    generator = pipeline('text-generation', model='titleos/Seahorse-350m')
    return generator(prompt, max_length=100, temperature=temperature)[0]['generated_text']

interface = gr.Interface(
    fn=generate_text,
    inputs=[
        gr.inputs.Textbox(lines=1, label="Instruction", default="### Instruction What are the three primary colors? ### Response"),
        gr.inputs.Slider(minimum=0.1, maximum=1.0, step=0.1, default=0.5, label="Temperature")
    ],
    outputs=gr.outputs.Textbox(label="Response")
)

interface.launch()