Sandy0909 commited on
Commit
26e4bac
·
1 Parent(s): ed1cde6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ class Config:
12
  class FinancialBERT(torch.nn.Module):
13
  def __init__(self):
14
  super(FinancialBERT, self).__init__()
15
- self.bert = BertForSequenceClassification.from_pretrained(Config.MODEL_PATH, num_labels=3, hidden_dropout_prob=0.5,model_file="model.bin")
16
 
17
  def forward(self, input_ids, attention_mask, labels=None):
18
  output = self.bert(input_ids, attention_mask=attention_mask, labels=labels)
 
12
  class FinancialBERT(torch.nn.Module):
13
  def __init__(self):
14
  super(FinancialBERT, self).__init__()
15
+ self.bert = BertForSequenceClassification.from_pretrained(Config.MODEL_PATH, num_labels=3, hidden_dropout_prob=0.5)
16
 
17
  def forward(self, input_ids, attention_mask, labels=None):
18
  output = self.bert(input_ids, attention_mask=attention_mask, labels=labels)