pipe = pipeline("fill-mask", model="DhimanBose/FineTuned_BanglaBERT_for_Masked_Language_Model") def image_classifier(inp): dat=pipe(inp) suggested_word=[] for x in dat: suggested_word.append(x['token_str']+" "+str(x['score'])) return 'Word'+" "+'Perchantage\n'+'\n '.join(suggested_word) demo = gr.Interface(fn=image_classifier,inputs="text", outputs="text") demo.launch()