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

trigger pipeline

Browse files
Files changed (1) hide show
  1. main.py +10 -12
main.py CHANGED
@@ -6,29 +6,27 @@ import logging
6
  import aiohttp
7
  from fastapi import FastAPI, File, UploadFile, HTTPException
8
  from fastapi.responses import JSONResponse
9
- from transformers import pipeline
10
- from transformers.pipelines import PipelineException
 
 
 
 
 
 
 
 
11
  from transformers import AutoImageProcessor, ViTForImageClassification
12
  from PIL import Image
13
  from cachetools import Cache
14
  import torch
15
  import torch.nn.functional as F
16
- import os
17
  from models import (
18
  FileImageDetectionResponse,
19
  UrlImageDetectionResponse,
20
  ImageUrlsRequest,
21
  )
22
 
23
- PATH = '/home/ahmadzen/.cache/huggingface/'
24
- DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests/'
25
- MODELPATH = '/home/ahmadzen/.cache/huggingface/model'
26
- os.environ['TRANSFORMERS_CACHE'] = MODELPATH
27
- os.environ['HF_HOME'] = PATH
28
- os.environ['HF_DATASETS_CACHE'] = DATASETPATH
29
- os.environ['TORCH_HOME'] = PATH
30
- os.environ['HF_HUB_CACHE'] = '/home/ahmadzen/.cache/huggingface/hub/'
31
-
32
  app = FastAPI()
33
  logging.basicConfig(
34
  level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
 
6
  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
21
  from cachetools import Cache
22
  import torch
23
  import torch.nn.functional as F
 
24
  from models import (
25
  FileImageDetectionResponse,
26
  UrlImageDetectionResponse,
27
  ImageUrlsRequest,
28
  )
29
 
 
 
 
 
 
 
 
 
 
30
  app = FastAPI()
31
  logging.basicConfig(
32
  level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"