IbrahimHasani commited on
Commit
c8fb72d
·
1 Parent(s): 8805736

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ CLIP_LEN = 32
10
 
11
  # Check if GPU is available and set the device
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
13
 
14
  # Load model and processor once and move them to the device
15
  processor = AutoProcessor.from_pretrained(MODEL_NAME)
 
10
 
11
  # Check if GPU is available and set the device
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
+ print ("device")
14
 
15
  # Load model and processor once and move them to the device
16
  processor = AutoProcessor.from_pretrained(MODEL_NAME)