diff --git a/requirements.txt b/requirements.txt index 55f2db8..99107e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,4 +12,5 @@ transformers jinja2==3.1.0 python-dotenv==1.0.1 PyJWT==2.9.0 -stripe \ No newline at end of file +stripe +sentry-sdk[fastapi] \ No newline at end of file diff --git a/services/image_generation_service.py b/services/image_generation_service.py index 28b4e21..9dcee52 100644 --- a/services/image_generation_service.py +++ b/services/image_generation_service.py @@ -20,6 +20,11 @@ from utils.data_types import Prompt, TextPrompt, TextToImage, ImageToImage, ValidatorInfo, ChatCompletion from fastapi.middleware.cors import CORSMiddleware from transformers import AutoTokenizer +import sentry_sdk + +sentry_sdk.init( + dsn="https://c29179b17e940dd744ef5987e54076ff@o4508692980170752.ingest.us.sentry.io/4508721596334080", +) # Define a list of allowed origins (domains) allowed_origins = [