Update

app.py

@@ -11,8 +11,8 @@ from pdb import set_trace as bp
 
 model = DemoModel()
 examples = json.load(open('amzn_examples.json'))
-cache = {x['text']: {'label': x['label']} for x in examples}
-
+# cache = {x['text']: {'label': x['label']} for x in examples}
+cache = json.load(open('cache.json'))
 unseen_labels = {x.strip() for x in open('cleaned_code/datasets/Amzn13K/unseen_labels_split6500_2.txt')}
 all_labels = {x.strip() for x in open('cleaned_code/datasets/Amzn13K/all_labels.txt')}
 
@@ -55,6 +55,7 @@ def classify(text, is_unseen):
         cache[text] = dict()
     cache[text]['preds'] = preds 
     cache[text]['descs'] = descs 
+    json.dump(cache, open('cache.json','w'), indent = 2)
 
     print(text, preds)
     # return preds
@@ -84,6 +85,9 @@ def toggle_descriptions_fn():
     descriptions_visible = not descriptions_visible
     return descriptions_visible
 
+# for example in examples:
+#     classify(example['text'], False)
+
 
 with gr.Blocks(css="#warning {height: 100%}") as demo:
     with gr.Column():