drhead commited on
Commit
c41c5f2
1 Parent(s): ca8e8cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -128,7 +128,7 @@ model.eval()
128
 
129
  with open("tagger_tags.json", "r") as file:
130
  tags = json.load(file) # type: dict
131
- allowed_tags = tags.keys()
132
 
133
  @spaces.GPU(duration=5)
134
  def create_tags(image, threshold):
 
128
 
129
  with open("tagger_tags.json", "r") as file:
130
  tags = json.load(file) # type: dict
131
+ allowed_tags = list(tags.keys())
132
 
133
  @spaces.GPU(duration=5)
134
  def create_tags(image, threshold):