juanmackie commited on
Commit
f95c930
·
verified ·
1 Parent(s): 2ed0f58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import numpy as np
4
  import cv2
5
  from PIL import Image
6
  import tempfile # For creating temporary video files
 
7
 
8
  # Marigold specific imports
9
  from diffusers import MarigoldDepthPipeline, DDIMScheduler
@@ -13,7 +14,7 @@ from huggingface_hub import login # For Hugging Face Hub login if needed
13
  CHECKPOINT = "prs-eth/marigold-depth-v1-1"
14
 
15
  # Check for HF_TOKEN_LOGIN environment variable for private models or higher rate limits
16
- if "HF_TOKEN_LOGIN" in os.environ: # os is imported implicitly by the Gradio environment, but good to add if running standalone
17
  login(token=os.environ["HF_TOKEN_LOGIN"])
18
 
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
4
  import cv2
5
  from PIL import Image
6
  import tempfile # For creating temporary video files
7
+ import os # <--- ADDED: Import the 'os' module
8
 
9
  # Marigold specific imports
10
  from diffusers import MarigoldDepthPipeline, DDIMScheduler
 
14
  CHECKPOINT = "prs-eth/marigold-depth-v1-1"
15
 
16
  # Check for HF_TOKEN_LOGIN environment variable for private models or higher rate limits
17
+ if "HF_TOKEN_LOGIN" in os.environ:
18
  login(token=os.environ["HF_TOKEN_LOGIN"])
19
 
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")