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)
|