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"
|