Jforeverss commited on
Commit
9fdc6e6
·
1 Parent(s): 836b4e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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