Spaces:
Runtime error
Runtime error
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() |