benstaf commited on
Commit
96d1185
·
verified ·
1 Parent(s): 2a18073

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,4 +1,7 @@
1
  import os
 
 
 
2
  from fastapi import FastAPI, UploadFile, File
3
  from transformers import AutoModelForImageClassification, AutoImageProcessor
4
  from PIL import Image
@@ -6,9 +9,6 @@ import torch.nn.functional as F
6
  import torch
7
  import io
8
 
9
- # ✅ Fix: set writable cache directory
10
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
11
-
12
  app = FastAPI()
13
 
14
  model = AutoModelForImageClassification.from_pretrained("prithivMLmods/Gender-Classifier-Mini")
 
1
  import os
2
+ # ✅ MUST be before importing transformers
3
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
4
+
5
  from fastapi import FastAPI, UploadFile, File
6
  from transformers import AutoModelForImageClassification, AutoImageProcessor
7
  from PIL import Image
 
9
  import torch
10
  import io
11
 
 
 
 
12
  app = FastAPI()
13
 
14
  model = AutoModelForImageClassification.from_pretrained("prithivMLmods/Gender-Classifier-Mini")