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=100, temperature=temperature)[0]['generated_text']
    response = generated_text.split("### Response")[-1]
    return response

interface = gr.Interface(
    fn=generate_text,
    inputs=[
        gr.inputs.Textbox(lines=1, label="Instruction", default="What are the three primary colors?"),
        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()