fffiloni commited on
Commit
c9af0cc
1 Parent(s): e2d1a94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -33,11 +33,9 @@ sys.path.append('src/blip')
33
  sys.path.append('clip-interrogator')
34
 
35
  import gradio as gr
36
- import spaces
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
@@ -46,7 +44,6 @@ config.blip_num_beams = 64
46
  ci = Interrogator(config)
47
 
48
 
49
- @spaces.GPU
50
  def inference(image, mode, best_max_flavors):
51
 
52
 
 
33
  sys.path.append('clip-interrogator')
34
 
35
  import gradio as gr
 
36
  from clip_interrogator import Config, Interrogator
37
 
38
  config = Config()
 
39
  config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
40
  config.blip_offload = False if torch.cuda.is_available() else True
41
  config.chunk_size = 2048
 
44
  ci = Interrogator(config)
45
 
46
 
 
47
  def inference(image, mode, best_max_flavors):
48
 
49