MUmairAB commited on
Commit
9ced299
·
1 Parent(s): 42ac6d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ my_interface = gr.Interface(title="Masked Language Model APP\n(by Umair Akram)",
33
  description="This App uses a fine-tuned DistilBERT-Base-Uncased Masked Language Model to predict the missed word in a sentence.\nEnter your text and put \"[MASK]\" at the word which you want to predict, as shown in the following example: Can we [MASK] to Paris?",
34
  fn=fill_the_mask,
35
  inputs="text",
36
- outputs="label")
37
 
38
  #Define the main function
39
  if __name__ == "__main__":
 
33
  description="This App uses a fine-tuned DistilBERT-Base-Uncased Masked Language Model to predict the missed word in a sentence.\nEnter your text and put \"[MASK]\" at the word which you want to predict, as shown in the following example: Can we [MASK] to Paris?",
34
  fn=fill_the_mask,
35
  inputs="text",
36
+ outputs="text")
37
 
38
  #Define the main function
39
  if __name__ == "__main__":