File size: 408 Bytes
b935e3d f404d24 b935e3d f404d24 b935e3d f404d24 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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()
|