cbensimon HF Staff commited on
Commit
b8d429b
·
1 Parent(s): 87359d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -42,6 +42,12 @@ 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
  ci = Interrogator(config)
46
 
47
  @spaces.GPU
 
42
  config.chunk_size = 2048
43
  config.flavor_intermediate_count = 512
44
  config.blip_num_beams = 64
45
+
46
+ @spaces.GPU
47
+ def cache_features
48
+ Interrogator(config)
49
+
50
+ cache_features()
51
  ci = Interrogator(config)
52
 
53
  @spaces.GPU