Spaces:
Runtime error
Runtime error
Commit
·
e30b067
1
Parent(s):
065e758
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ class Bert_model(nn.Module):
|
|
38 |
return logits
|
39 |
|
40 |
# Load the model
|
41 |
-
model_path = "
|
42 |
state_dict = torch.load(model_path)
|
43 |
device = "cuda" # or "cpu" if GPU is not available
|
44 |
|
|
|
38 |
return logits
|
39 |
|
40 |
# Load the model
|
41 |
+
model_path = "model.pt" # Replace with your actual model path
|
42 |
state_dict = torch.load(model_path)
|
43 |
device = "cuda" # or "cpu" if GPU is not available
|
44 |
|