Upload app.py

app.py

@@ -13,6 +13,9 @@ class_model=torch.load('models/class.pt')
 
 
 def greet(name):
+
+    translation_table = str.maketrans("", "", " \t\n\r\f\v")
+    name = name.translate(translation_table)
     token_ids = torch.tensor([tokenizer.encode(name)])
     token_ids = token_ids
     bert_output = bert_model(token_ids)