sandrocalzada commited on
Commit
8006c7e
1 Parent(s): bbb8961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ app.prepare(ctx_id=0, det_size=(640, 640))
13
  swapper = insightface.model_zoo.get_model(os.path.join(os.getcwd(), 'inswapper_128.onnx'), download=False)
14
 
15
  # Load pickle file once and keep it in memory
16
- @st.cache(allow_output_mutation=True)
17
  def load_data_images():
18
  with open(os.path.join(os.getcwd(), 'data_images.pkl'), 'rb') as file:
19
  return pickle.load(file)
 
13
  swapper = insightface.model_zoo.get_model(os.path.join(os.getcwd(), 'inswapper_128.onnx'), download=False)
14
 
15
  # Load pickle file once and keep it in memory
16
+ @st.cache_data
17
  def load_data_images():
18
  with open(os.path.join(os.getcwd(), 'data_images.pkl'), 'rb') as file:
19
  return pickle.load(file)