Update main.py
Browse files
    	
        main.py
    CHANGED
    
    | @@ -2,7 +2,7 @@ from fastapi import FastAPI, HTTPException | |
| 2 | 
             
            from pydantic import BaseModel, root_validator
         | 
| 3 | 
             
            from transformers import AutoModel
         | 
| 4 | 
             
            from typing import List
         | 
| 5 | 
            -
            import os
         | 
| 6 |  | 
| 7 | 
             
            if platform.system() == "Windows":
         | 
| 8 | 
             
                print("Windows detected. Assigning cache directory to Transformers in AppData\Local.")
         | 
|  | |
| 2 | 
             
            from pydantic import BaseModel, root_validator
         | 
| 3 | 
             
            from transformers import AutoModel
         | 
| 4 | 
             
            from typing import List
         | 
| 5 | 
            +
            import os, platform
         | 
| 6 |  | 
| 7 | 
             
            if platform.system() == "Windows":
         | 
| 8 | 
             
                print("Windows detected. Assigning cache directory to Transformers in AppData\Local.")
         |