trigger pipeline

main.py

@@ -6,29 +6,27 @@ import logging
 import aiohttp
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import JSONResponse
-from transformers import pipeline
-from transformers.pipelines import PipelineException
+import os
+PATH = '/home/ahmadzen/.cache/huggingface/'
+DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests/'
+MODELPATH = '/home/ahmadzen/.cache/huggingface/model/'
+os.environ['TRANSFORMERS_CACHE'] = MODELPATH
+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
 from cachetools import Cache
 import torch
 import torch.nn.functional as F
-import os
 from models import (
     FileImageDetectionResponse,
     UrlImageDetectionResponse,
     ImageUrlsRequest,
 )
 
-PATH = '/home/ahmadzen/.cache/huggingface/'
-DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests/'
-MODELPATH = '/home/ahmadzen/.cache/huggingface/model'
-os.environ['TRANSFORMERS_CACHE'] = MODELPATH
-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/'
-
 app = FastAPI()
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"