Update app.py

app.py

@@ -71,7 +71,10 @@ input_size = 768 + metadata_features  # featurizer output size + metadata size
 log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLASSES)
 
 # Load Weights
-model_weights_path = 'gena-blastln-bs33-lr4e-05-S168.pth'
+import os
+
+# Get the model path from the environment variable
+model_path = os.getenv('MODEL_PATH')
 weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
 
 base_model.load_state_dict(weights['model_state_dict'])