asahi417 commited on
Commit
35e486f
1 Parent(s): 3a0722f

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ css = """
14
  @spaces.GPU
15
  def infer(image, best_max_flavors):
16
  image = image.convert('RGB')
17
- return model.image2text([image], best_max_flavors=best_max_flavors)[0]
18
 
19
 
20
  with gr.Blocks(css=css) as demo:
 
14
  @spaces.GPU
15
  def infer(image, best_max_flavors):
16
  image = image.convert('RGB')
17
+ return model(image, best_max_flavors=best_max_flavors)
18
 
19
 
20
  with gr.Blocks(css=css) as demo:
requirements.txt CHANGED
@@ -5,4 +5,4 @@ torch
5
  torchvision
6
  spaces
7
  clip-interrogator
8
- panna>=0.0.4
 
5
  torchvision
6
  spaces
7
  clip-interrogator
8
+ panna>=0.1.0