import gradio as gr
from transformers import pipeline

pipeline = pipeline(task="text-generation", model="meta-llama/Llama-2-7b")

def predict(input):
    return pipeline("I can't believe you did such a ", do_sample=False)

gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=2, placeholder="Text here..."),
    outputs="text",
).launch()