Spaces:
Runtime error
Runtime error
Commit
·
9fdc6e6
1
Parent(s):
836b4e0
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ class Bert_model(nn.Module):
|
|
35 |
return logits
|
36 |
|
37 |
# Load the model
|
38 |
-
model_path = "finchat222/model.pt" # Replace with your actual model path
|
39 |
state_dict = torch.load(model_path)
|
40 |
device = "cuda" # or "cpu" if GPU is not available
|
41 |
|
|
|
35 |
return logits
|
36 |
|
37 |
# Load the model
|
38 |
+
model_path = "Jforeverss/finchat222/model.pt" # Replace with your actual model path
|
39 |
state_dict = torch.load(model_path)
|
40 |
device = "cuda" # or "cpu" if GPU is not available
|
41 |
|