app.py

import gradio as gr
from transformers import pipeline

# Use a pipeline as a high-level helper
pipe = pipeline("text-generation", model="meta-llama/Llama-2-7b-chat-hf")

def llama_chat(name):
    resp = pipe(name)
    return resp

iface = gr.Interface(fn=llama_chat, inputs="text", outputs="text")
iface.launch()