aznasut commited on
Commit
4384221
·
1 Parent(s): ce0e37a

trigger pipeline

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -7,14 +7,14 @@ import aiohttp
7
  from fastapi import FastAPI, File, UploadFile, HTTPException
8
  from fastapi.responses import JSONResponse
9
  import os
10
- PATH = '/home/ahmadzen/.cache/huggingface/'
11
- DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests/'
12
- MODELPATH = '/home/ahmadzen/.cache/huggingface/model/'
13
  os.environ['TRANSFORMERS_CACHE'] = MODELPATH
14
  os.environ['HF_HOME'] = PATH
15
  os.environ['HF_DATASETS_CACHE'] = DATASETPATH
16
  os.environ['TORCH_HOME'] = PATH
17
- os.environ['HF_HUB_CACHE'] = '/home/ahmadzen/.cache/huggingface/hub/'
18
 
19
  from transformers import AutoImageProcessor, ViTForImageClassification
20
  from PIL import Image
 
7
  from fastapi import FastAPI, File, UploadFile, HTTPException
8
  from fastapi.responses import JSONResponse
9
  import os
10
+ PATH = '/home/ahmadzen/.cache/huggingface'
11
+ DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests'
12
+ MODELPATH = '/home/ahmadzen/.cache/huggingface/model'
13
  os.environ['TRANSFORMERS_CACHE'] = MODELPATH
14
  os.environ['HF_HOME'] = PATH
15
  os.environ['HF_DATASETS_CACHE'] = DATASETPATH
16
  os.environ['TORCH_HOME'] = PATH
17
+ os.environ['HF_HUB_CACHE'] = '/home/ahmadzen/.cache/huggingface/hub'
18
 
19
  from transformers import AutoImageProcessor, ViTForImageClassification
20
  from PIL import Image