File size: 259 Bytes
bf22482 7c263a3 bf22482 7c263a3 bf22482 7c263a3 |
1 2 3 4 5 6 7 8 |
import gradio as gr
from transformers import pipeline, GenerationConfig
pipe = pipeline("text-generation", model="tpzeng/prm800k_llama_joint_checkpoint1000", config=GenerationConfig(bad_words_ids=[128254,128255]))
gr.Interface.from_pipeline(pipe).launch()
|