saurabhg2083 commited on
Commit
0f38a56
1 Parent(s): dbd7e4b
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,12 +9,12 @@ from happytransformer import HappyTextToText, TTSettings
9
  import torch
10
  from torch.nn import BCEWithLogitsLoss
11
  from torch.utils.data import DataLoader, TensorDataset, random_split
 
12
 
13
- pipe = pipeline("text-classification", model="saurabhg2083/model_bert")
 
14
  happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
15
  args = TTSettings(num_beams=5, min_length=1)
16
- model = BertForSequenceClassification.from_pretrained(save_directory)
17
- tokenizer = BertTokenizer.from_pretrained(save_directory)
18
 
19
  gendered_pronouns = [
20
  'ambition', 'driven', 'lead', 'persist', 'principle', 'decision', 'superior', 'individual', 'assertive',
 
9
  import torch
10
  from torch.nn import BCEWithLogitsLoss
11
  from torch.utils.data import DataLoader, TensorDataset, random_split
12
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("saurabhg2083/model_bert")
15
+ model = AutoModelForSequenceClassification.from_pretrained("saurabhg2083/model_bert")
16
  happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
17
  args = TTSettings(num_beams=5, min_length=1)
 
 
18
 
19
  gendered_pronouns = [
20
  'ambition', 'driven', 'lead', 'persist', 'principle', 'decision', 'superior', 'individual', 'assertive',