import gradio as gr
from transformers import pipeline

# Load a simple model (fill-mask task as an example)
model = pipeline("fill-mask", model="distilroberta-base")

# Define the function to use the GPU
def predict(text):
    return model(text)

# Gradio Interface
interface = gr.Interface(fn=predict, inputs="text", outputs="text")

# Launch the interface
if __name__ == "__main__":
    interface.launch()