fffiloni commited on
Commit
b28dbb0
1 Parent(s): f99eac6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,12 +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
-
46
 
47
 
48
  @spaces.GPU
49
  def inference(image, mode, best_max_flavors):
50
- ci = Interrogator(config)
51
 
52
  image = image.convert('RGB')
53
  if mode == 'best':
 
42
  config.chunk_size = 2048
43
  config.flavor_intermediate_count = 512
44
  config.blip_num_beams = 64
45
+ ci = Interrogator(config)
46
 
47
 
48
  @spaces.GPU
49
  def inference(image, mode, best_max_flavors):
50
+
51
 
52
  image = image.convert('RGB')
53
  if mode == 'best':