Chenyu Yang
llama2
18d5051
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()