fffiloni commited on
Commit
e2d1a94
1 Parent(s): 63bee0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -37,13 +37,12 @@ import spaces
37
  from clip_interrogator import Config, Interrogator
38
 
39
  config = Config()
 
40
  config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
41
  config.blip_offload = False if torch.cuda.is_available() else True
42
  config.chunk_size = 2048
43
  config.flavor_intermediate_count = 512
44
  config.blip_num_beams = 64
45
-
46
- @spaces.GPU
47
  ci = Interrogator(config)
48
 
49
 
 
37
  from clip_interrogator import Config, Interrogator
38
 
39
  config = Config()
40
+ @spaces.GPU
41
  config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
42
  config.blip_offload = False if torch.cuda.is_available() else True
43
  config.chunk_size = 2048
44
  config.flavor_intermediate_count = 512
45
  config.blip_num_beams = 64
 
 
46
  ci = Interrogator(config)
47
 
48