update
This commit is contained in:
parent
bf34ce90c5
commit
d0114f8700
3
main.py
3
main.py
@ -28,7 +28,7 @@ if not GEMINI_API_KEY:
|
||||
raise RuntimeError("GEMINI_API_KEY not set")
|
||||
|
||||
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")
|
||||
|
||||
@ -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
|
||||
the same person, False otherwise.
|
||||
"""
|
||||
print(f"[DEBUG] Identity check using model: {IDENTITY_MODEL}")
|
||||
try:
|
||||
original_part = Part.from_bytes(
|
||||
data=original_bytes,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user