Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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'])
|