steelcityamir/safe-content-ai
1
1
Fork 0
mirror of https://github.com/steelcityamir/safe-content-ai.git synced 2024-09-11 18:37:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Boroumand, Amir A 2024-04-22 14:02:01 -04:00
parent e68434fb94
commit 7c27abe4bc
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@ -5,7 +5,8 @@ import hashlib
import logging
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import JSONResponse
from transformers import pipeline, PipelineException
from transformers import pipeline
from transformers.pipelines import PipelineException
from PIL import Image
from cachetools import Cache