Update app.py
Browse files
app.py
CHANGED
@@ -51,8 +51,7 @@ def predict(image: Image.Image):
|
|
51 |
|
52 |
scores = {tags[i]: result[i] for i in range(len(result))}
|
53 |
predicted_tags = [tag for tag, score in scores.items() if score > THRESHOLD]
|
54 |
-
tag_string = ', '.join(predicted_tags)
|
55 |
-
tag_string = str.tag_string.replace("_", " ").replace("(", "\(").replace(")", "\)")
|
56 |
|
57 |
return tag_string, scores
|
58 |
|
|
|
51 |
|
52 |
scores = {tags[i]: result[i] for i in range(len(result))}
|
53 |
predicted_tags = [tag for tag, score in scores.items() if score > THRESHOLD]
|
54 |
+
tag_string = ', '.join(predicted_tags).replace("_", " ").replace("(", "\(").replace(")", "\)")
|
|
|
55 |
|
56 |
return tag_string, scores
|
57 |
|