ayushsinha's picture
Upload 2 files
9c69e6d verified
raw
history blame contribute delete
714 Bytes
import gradio as gr
from transformers import pipeline
def fill_mask(text):
model_name = "AventIQ-AI/RoBERTa"
mask_filler = pipeline("fill-mask", model=model_name)
results = mask_filler(text)
output = "\n".join([f"{i+1}. {res['sequence']} (Confidence: {res['score']:.4f})" for i, res in enumerate(results)])
return output
iface = gr.Interface(
fn=fill_mask,
inputs=gr.Textbox(label="Input Text (Use <mask> for missing word)"),
outputs=gr.Textbox(label="Predictions"),
title="RoBERTa Fill Mask",
description="Enter a sentence with <mask>, and the RoBERTa model will predict the missing word."
)
if __name__ == "__main__":
iface.launch()