Update config.py
Browse files
    	
        config.py
    CHANGED
    
    | @@ -9,7 +9,7 @@ EPOCHS = 5 | |
| 9 |  | 
| 10 | 
             
            # Folder to contain all the datasets
         | 
| 11 | 
             
            DATASET_LOCATION = "" # 
         | 
| 12 | 
            -
             | 
| 13 |  | 
| 14 | 
             
            device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         | 
| 15 |  | 
|  | |
| 9 |  | 
| 10 | 
             
            # Folder to contain all the datasets
         | 
| 11 | 
             
            DATASET_LOCATION = "" # 
         | 
| 12 | 
            +
            MODEL_PATH = "thak123/bert-emoji-latvian-twitter-classifier/model.bin"
         | 
| 13 |  | 
| 14 | 
             
            device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         | 
| 15 |  |