create app file

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from transformers import BartTokenizer, BartForConditionalGeneration
+tokenizer = BartTokenizer.from_pretrained("facebook/bart-base")
+model = BartForConditionalGeneration.from_pretrained("facebook/bart-base")
+def predict(text, number of predictions):
+  TXT = text + "<mask>"
+  input_ids = tokenizer([TXT], return_tensors="pt")["input_ids"]
+  logits = model(input_ids).logits
+  
+  masked_index = (input_ids[0] == tokenizer.mask_token_id).nonzero().item()
+  probs = logits[0, masked_index].softmax(dim=0)
+  values, predictions = probs.topk(5)
+  list_of_words = tokenizer.decode(predictions).split()
+  return ', '.join([str(elem) for elem in list_of_words])
+intr = gr.Interface(predict, ["text","number"], "text")
+intr.launch(inline = False)