IbrahimHasani commited on
Commit
ba82304
·
1 Parent(s): f69a7a0

Update app.py

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