Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
5e187cb
1
Parent(s):
c8d73ef
Update app.py
Browse files
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"]
|