Spaces:
Sleeping
Sleeping
Commit
·
0f0bf1b
1
Parent(s):
06247a9
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ bert_model = BertForSequenceClassification.from_pretrained('my_bert_model')
|
|
23 |
tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
|
24 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
25 |
bert_model = bert_model.to(device)
|
26 |
-
model_finetuned =
|
27 |
model_finetuned.eval()
|
28 |
|
29 |
labels = ["не токсичный", "оскорбляющий", "непристойный", "угрожающий", "опасный"]
|
|
|
23 |
tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
|
24 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
25 |
bert_model = bert_model.to(device)
|
26 |
+
model_finetuned = torch.load('GPT_2')
|
27 |
model_finetuned.eval()
|
28 |
|
29 |
labels = ["не токсичный", "оскорбляющий", "непристойный", "угрожающий", "опасный"]
|