Spaces:

fffiloni
/

CLIP-Interrogator-2

Running on T4

fffiloni commited on Dec 13, 2023

Commit

b28dbb0

1 Parent(s): f99eac6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,12 +42,12 @@ config.blip_offload = False if torch.cuda.is_available() else True
 config.chunk_size = 2048
 config.flavor_intermediate_count = 512
 config.blip_num_beams = 64
 @spaces.GPU
 def inference(image, mode, best_max_flavors):
-    ci = Interrogator(config)
     image = image.convert('RGB')
     if mode == 'best':

 config.chunk_size = 2048
 config.flavor_intermediate_count = 512
 config.blip_num_beams = 64
+ci = Interrogator(config)
 @spaces.GPU
 def inference(image, mode, best_max_flavors):
     image = image.convert('RGB')
     if mode == 'best':