Update app.py

app.py

@@ -7,11 +7,12 @@ from PIL import Image
 import json
 import zipfile
 import pickle
+import pickletools
 from transformers import AutoTokenizer, CLIPTextModelWithProjection
 
 # loading the train dataset
 with open('clip_train.pkl', 'rb') as f: 
-    temp_d = pickle.load(f)
+    temp_d = pickletools.dis(f)
     # train_xv = temp_d['image'].astype(np.float64)   # Array of image features : np ndarray
     # train_xt = temp_d['text'].astype(np.float64)    # Array of text features : np ndarray
     # train_yv  = temp_d['label']                     # Array of labels 
@@ -20,7 +21,7 @@ with open('clip_train.pkl', 'rb') as f:
 
 # loading the test dataset
 with open('clip_test.pkl', 'rb') as f:
-    temp_d = pickle.load(f)
+    temp_d = pickletools.dis(f)
     # test_xv = temp_d['image'].astype(np.float64)
     test_xt = temp_d['text'].astype(np.float64)
     # test_yv = temp_d['label']