LPX55 commited on
Commit
c2d983a
·
verified ·
1 Parent(s): ee9e061

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,13 +3,13 @@ import os
3
  import torch
4
  import numpy as np
5
  import cv2
 
6
  import matplotlib.pyplot as plt
7
  from PIL import Image
8
  from sam2.build_sam import build_sam2
9
  from sam2.sam2_image_predictor import SAM2ImagePredictor
10
 
11
 
12
-
13
  # Remove all CUDA-specific configurations
14
  torch.autocast(device_type="cpu", dtype=torch.float32).__enter__()
15
 
 
3
  import torch
4
  import numpy as np
5
  import cv2
6
+ import huggingface_hub
7
  import matplotlib.pyplot as plt
8
  from PIL import Image
9
  from sam2.build_sam import build_sam2
10
  from sam2.sam2_image_predictor import SAM2ImagePredictor
11
 
12
 
 
13
  # Remove all CUDA-specific configurations
14
  torch.autocast(device_type="cpu", dtype=torch.float32).__enter__()
15