# 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")