import gradio as gr
from transformers import pipeline

# Initialize the text generation pipeline
pipe = pipeline("text-generation", model="meta-llama/CodeLlama-7b-hf")

def generate_text(prompt):
    # Generate text based on the provided prompt using the pipeline
    generated_text = pipe(prompt, max_length=50, do_sample=True)[0]['generated_text']
    return generated_text

# Create a Gradio interface
demo = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Text Generation with CodeLlama-7b-hf",
    description="Enter a prompt and see the generated text."
)

# Launch the Gradio interface
demo.launch()