Add model card
Browse files
    	
        README.md
    CHANGED
    
    | @@ -56,7 +56,10 @@ import torch | |
| 56 |  | 
| 57 | 
             
            # Download model
         | 
| 58 | 
             
            model_path = hf_hub_download("munchers/scent-to-molecule", "best_control.pt")
         | 
| 59 | 
            -
             | 
|  | |
|  | |
|  | |
| 60 | 
             
            ```
         | 
| 61 |  | 
| 62 | 
             
            ## Examples
         | 
|  | |
| 56 |  | 
| 57 | 
             
            # Download model
         | 
| 58 | 
             
            model_path = hf_hub_download("munchers/scent-to-molecule", "best_control.pt")
         | 
| 59 | 
            +
             | 
| 60 | 
            +
            model.load_state_dict(checkpoint['model_state_dict'])
         | 
| 61 | 
            +
            if torch.cuda.is_available():
         | 
| 62 | 
            +
                model = model.cuda()
         | 
| 63 | 
             
            ```
         | 
| 64 |  | 
| 65 | 
             
            ## Examples
         |