Update app.py

app.py

@@ -1,6 +1,7 @@
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 import os, subprocess
+import spaces
 import torch
 
 def setup():
@@ -35,7 +36,7 @@ sys.path.append('clip-interrogator')
 import gradio as gr
 from clip_interrogator import Config, Interrogator
 
-config = Config()
+config = Config(download_cache=True)
 config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
 config.blip_offload = False if torch.cuda.is_available() else True
 config.chunk_size = 2048
@@ -43,7 +44,7 @@ config.flavor_intermediate_count = 512
 config.blip_num_beams = 64
 ci = Interrogator(config)
 
-
+@spaces.GPU
 def inference(image, mode, best_max_flavors):