Spaces:
Runtime error
Runtime error
Commit
·
120c541
1
Parent(s):
a81a078
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,8 +7,8 @@ device = torch.device('cpu')
|
|
| 7 |
|
| 8 |
NUM_CLASSES=6
|
| 9 |
|
| 10 |
-
model=BertForMaskedLM.from_pretrained("
|
| 11 |
-
tokenizer=BertTokenizer.from_pretrained("
|
| 12 |
|
| 13 |
|
| 14 |
def predict(text=None) -> dict:
|
|
|
|
| 7 |
|
| 8 |
NUM_CLASSES=6
|
| 9 |
|
| 10 |
+
model=BertForMaskedLM.from_pretrained("./")
|
| 11 |
+
tokenizer=BertTokenizer.from_pretrained("./")
|
| 12 |
|
| 13 |
|
| 14 |
def predict(text=None) -> dict:
|