MatthiasC commited on
Commit
72a955a
1 Parent(s): 9f824d9

Model hub changed to ml6team

Browse files
Files changed (1) hide show
  1. server.py +2 -2
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="MatthiasC/dall-e-logo", filename="full_dict_new.ckpt",
25
- use_auth_token=st.secrets["model_hub"])
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"