Update app.py

app.py

@@ -44,10 +44,10 @@ from diffusers import UNet2DConditionModel
 def load_models():
     torch_device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
-    image_processor_model_path = os.path.join(settings.BASE_DIR, 'depthAPI', 'models', 'image_processor')
-    swin_transformer_model_path = os.path.join(settings.BASE_DIR, 'depthAPI', 'models', 'swin_transformer')
-    vae_model_path = os.path.join(settings.BASE_DIR, 'depthAPI', 'models', 'vae', 'MonoChannelVAE.pth')
-    unet_model_path = os.path.join(settings.BASE_DIR, 'depthAPI', 'models', 'unet')
+    image_processor_model_path = 'models/image_processor/image_processor'
+    swin_transformer_model_path = 'models/swin_transformer/swin_transformer'
+    vae_model_path = 'models/vae/vae/MonoChannelVAE.pth'
+    unet_model_path = 'models/unet/unet'
 
     image_processor = AutoImageProcessor.from_pretrained(image_processor_model_path)
     swin_transformer = SwinModel.from_pretrained(swin_transformer_model_path)