Update app.py

app.py

@@ -28,6 +28,10 @@ def load_tags_info():
             
             i += 1
     
+    tag_to_id['None'] = 155
+    id_to_tag['155'] = 'None'
+    id_to_description['155'] = 'No tag'
+    
     return (tag_to_id, id_to_tag, id_to_description)
 
 tag_to_id, id_to_tag, id_to_description = load_tags_info()
@@ -46,6 +50,8 @@ def top_xx(preds, xx=95):
     result = []
     while total < xx / 100:
         next_id = tops[0, index].item()
+        if next_id == 155:
+            continue
         total += preds[0, next_id]
         index += 1
         result.append({'tag': id_to_tag[next_id], 'description': id_to_description[next_id]})