Update app.py
Browse files
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):
|