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