Jforeverss commited on
Commit
e30b067
·
1 Parent(s): 065e758

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ class Bert_model(nn.Module):
38
  return logits
39
 
40
  # Load the model
41
- model_path = "finchat222/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
 
 
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