alfredplpl commited on
Commit
5e187cb
β€’
1 Parent(s): c8d73ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import torch
6
  import os
7
 
8
  from huggingface_hub import hf_hub_download
 
9
 
10
  model_id = "aipicasso/emi-2"
11
  token=os.environ["TOKEN"]
 
6
  import os
7
 
8
  from huggingface_hub import hf_hub_download
9
+ from safetensors.torch import load_file
10
 
11
  model_id = "aipicasso/emi-2"
12
  token=os.environ["TOKEN"]