Update app.py

app.py

@@ -14,10 +14,10 @@ class_model=torch.load('models/class.pt')
 
 def greet(name):
 
-    translation_table = str.maketrans("", "", " \t\n\r\f\v")
-    name = name.translate(translation_table)
+    #translation_table = str.maketrans("", "", " \t\n\r\f\v")
+    name = name.replace(" ", "").replace("\n", "").replace("\t", "")
     token_ids = torch.tensor([tokenizer.encode(name)])
-    token_ids = token_ids
+ 
     bert_output = bert_model(token_ids)
     class_output=class_model(bert_output[1])
     cluster = torch.argmax(class_output, dim=1) + 1