silveroxides commited on
Commit
f586952
·
verified ·
1 Parent(s): 0560975

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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