mawairon commited on
Commit
ba0faf4
·
verified ·
1 Parent(s): db9f444

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLAS
74
  import os
75
 
76
  # Get the model path from the environment variable
77
- model_path = os.getenv('MODEL_PATH')
78
  weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
79
 
80
  base_model.load_state_dict(weights['model_state_dict'])
 
74
  import os
75
 
76
  # Get the model path from the environment variable
77
+ model_weights_path = os.getenv('MODEL_PATH')
78
  weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
79
 
80
  base_model.load_state_dict(weights['model_state_dict'])