import gradio as gr
import transformers
from transformers import pipeline

def generate_text(prompt, temperature):
    generator = pipeline('text-generation', model='titleos/Seahorse-350m')
    generated_text = generator("### Instruction " + prompt + " ### Response ", max_length=250, temperature=temperature)[0]['generated_text']
    response = generated_text.split("### Response")[-1]
    return response

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

interface.launch()