Update app.py

app.py

@@ -4,6 +4,7 @@ import os.path
 
 import gradio as gr
 
+import pickle
 import tqdm
 import json