Spaces:
Build error
Build error
Model hub changed to ml6team
Browse files
server.py
CHANGED
@@ -21,8 +21,8 @@ app = FastAPI()
|
|
21 |
from huggingface_hub import hf_hub_download
|
22 |
|
23 |
logging.info("Start downloading")
|
24 |
-
full_dict_path = hf_hub_download(repo_id="
|
25 |
-
use_auth_token=st.secrets["
|
26 |
logging.info("End downloading")
|
27 |
|
28 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
21 |
from huggingface_hub import hf_hub_download
|
22 |
|
23 |
logging.info("Start downloading")
|
24 |
+
full_dict_path = hf_hub_download(repo_id="ml6team/logo-generator", filename="full_dict_new.ckpt",
|
25 |
+
use_auth_token=st.secrets["model_download"])
|
26 |
logging.info("End downloading")
|
27 |
|
28 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|