KeXing
commited on
Commit
•
4616fcc
1
Parent(s):
212111c
Upload app.py
Browse files
app.py
CHANGED
@@ -13,6 +13,9 @@ class_model=torch.load('models/class.pt')
|
|
13 |
|
14 |
|
15 |
def greet(name):
|
|
|
|
|
|
|
16 |
token_ids = torch.tensor([tokenizer.encode(name)])
|
17 |
token_ids = token_ids
|
18 |
bert_output = bert_model(token_ids)
|
|
|
13 |
|
14 |
|
15 |
def greet(name):
|
16 |
+
|
17 |
+
translation_table = str.maketrans("", "", " \t\n\r\f\v")
|
18 |
+
name = name.translate(translation_table)
|
19 |
token_ids = torch.tensor([tokenizer.encode(name)])
|
20 |
token_ids = token_ids
|
21 |
bert_output = bert_model(token_ids)
|