Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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)
|