gpu

app.py

@@ -25,7 +25,6 @@ models_info = [{"name": "ae_model_tf_2024-03-05_00-35-21.pth", "feature_dim": 32
         {"name": "autoencoder-epoch=49-train_loss=1.01.ckpt", "feature_dim": 64}]
 models = []
 
-@spaces.GPU
 def load_model(model_path, feature_dim):
     model = AutoencoderModule(feature_dim=feature_dim)
     state_dict = torch.load(model_path)