Test / app.py
ThomasSimonini's picture
Update app.py
62d0599
raw history blame
No virus
231 Bytes
import gradio as gr
from transformers import pipeline
hf_token = os.getenv("HF_TOKEN")
pipe = pipeline("conversational", model="meta-llama/Llama-2-7b-chat-hf", token=hf_token)
demo = gr.Interface.from_pipeline(pipe)
demo.launch()