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()