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