Update main.py

main.py

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModel
 import torch
 from torch import Tensor
 import torch.nn.functional as F
-
+import os 
 os.environ['HF_HOME'] = '/'
 
 app = FastAPI()