fazli/FastApi-GeminiApi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Hossein 2025-12-01 15:07:49 +03:30
parent bf34ce90c5
commit d0114f8700
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

@ -28,7 +28,7 @@ if not GEMINI_API_KEY:
raise RuntimeError("GEMINI_API_KEY not set") raise RuntimeError("GEMINI_API_KEY not set")
IMAGE_MODEL = "gemini-2.5-flash-image" IMAGE_MODEL = "gemini-2.5-flash-image"
IDENTITY_MODEL = "gemini-1.5-flash" # text/vision model for identity check IDENTITY_MODEL = "gemini-1.5-pro" # multimodal text+vision model for identity check
AUTH_KEY = os.getenv("AIFIT_AUTH_KEY") AUTH_KEY = os.getenv("AIFIT_AUTH_KEY")
@ -86,6 +86,7 @@ def faces_match_strict(original_bytes: bytes, edited_bytes: bytes) -> bool:
if identity/face has changed. Returns True if it's clearly if identity/face has changed. Returns True if it's clearly
the same person, False otherwise. the same person, False otherwise.
""" """
print(f"[DEBUG] Identity check using model: {IDENTITY_MODEL}")
try: try:
original_part = Part.from_bytes( original_part = Part.from_bytes(
data=original_bytes, data=original_bytes,