Test / app.py
ThomasSimonini's picture
Update app.py
028cf00
import os
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()