Vladislawoo commited on
Commit
0f0bf1b
·
1 Parent(s): 06247a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = load_full_model_from_pt(model_directory)
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 = ["не токсичный", "оскорбляющий", "непристойный", "угрожающий", "опасный"]