Spaces:

aznasut
/

ViT_Deepfake_Detection

Sleeping

aznasut commited on Oct 23, 2024

Commit

e11afaf

1 Parent(s): cf21eaf

trigger pipeline

Files changed (1) hide show

main.py CHANGED Viewed

@@ -13,7 +13,7 @@ DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests'
 os.environ['HF_HOME'] = PATH
 os.environ['HF_DATASETS_CACHE'] = DATASETPATH
 os.environ['TORCH_HOME'] = PATH
-os.environ['HF_HUB_CACHE'] = '/home/ahmadzen/.cache/huggingface/hub'
 from transformers import AutoImageProcessor, ViTForImageClassification
 from PIL import Image

 os.environ['HF_HOME'] = PATH
 os.environ['HF_DATASETS_CACHE'] = DATASETPATH
 os.environ['TORCH_HOME'] = PATH
+os.environ['HF_HUB_CACHE'] = '/home/ahmadzen/.cache/huggingface'
 from transformers import AutoImageProcessor, ViTForImageClassification
 from PIL import Image