Hugging Face's logo

Spaces:
gokaygokay
/
FLUX.1-dev-with-Captioner
Running on Zero

App Files Community
3
gokaygokay commited on
Commit
932af65
·
verified ·
1 Parent(s): edb9e43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ MAX_SEED = np.iinfo(np.int32).max
30
  MAX_IMAGE_SIZE = 2048
31
 
32
  # Florence caption function
 
33
  def florence_caption(image):
34
  # Convert image to PIL if it's not already
35
  if not isinstance(image, Image.Image):
 
30
  MAX_IMAGE_SIZE = 2048
31
 
32
  # Florence caption function
33
+ @spaces.GPU
34
  def florence_caption(image):
35
  # Convert image to PIL if it's not already
36
  if not isinstance(image, Image.Image):