Update app.py

app.py

@@ -7,8 +7,8 @@ device = torch.device('cpu')
 
 NUM_CLASSES=6
 
-model=BertForMaskedLM.from_pretrained("./")
-tokenizer=BertTokenizer.from_pretrained("./")
+#model=BertForMaskedLM.from_pretrained("./")
+#tokenizer=BertTokenizer.from_pretrained("./")
 
 
 def predict(text=None) -> dict:  
@@ -34,4 +34,4 @@ def test():
 gr.Interface(fn=test, 
              inputs=gr.inputs.Textbox(lines=2, placeholder="Your Text… "),
              title="Mask Language Modeling - Demo",
-             outputs=gr.outputs.Label(num_top_classes=NUM_CLASSES)).launch()
+             outputs=gr.outputs.Label(num_top_classes=1)).launch()