| # Use a pipeline as a high-level helper | |
| from transformers import pipeline | |
| pipe = pipeline("text-generation", model="ibm-granite/granite-8b-code-instruct", device=0) | |
| import gradio as gr | |
| def prompt_model(prompt): | |
| return pipe(prompt) | |
| demo = gr.Interface(fn=prompt_model, inputs="textbox", outputs="textbox") |