Update app.py

app.py

@@ -6,7 +6,7 @@ import base64
 from io import BytesIO
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-import imagehash  # <-- NOVO
+import imagehash
 
 MODEL_NAME = "facebook/dinov2-small"
 processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
@@ -21,6 +21,7 @@ app = FastAPI(
 
 class ImageRequest(BaseModel):
     image: str
+    use_float16: bool = False # <-- NOVO: Parâmetro opcional com valor padrão False
 
 @app.post("/embed")
 async def get_embedding(request: ImageRequest):
@@ -34,10 +35,15 @@ async def get_embedding(request: ImageRequest):
             last_hidden_state = outputs.last_hidden_state
             embedding = last_hidden_state[:, 0]
             embedding_512 = projection(embedding)
-        phash = str(imagehash.phash(image))  # <-- NOVO
+
+        # <-- NOVA LÓGICA: Conversão condicional para float16
+        if request.use_float16:
+            embedding_512 = embedding_512.half()
+
+        phash = str(imagehash.phash(image))
         return {
             "embedding": embedding_512.squeeze().tolist(),
             "phash": phash
         }
     except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Erro ao processar a imagem: {e}")
+        raise HTTPException(status_code=400, detail=f"Erro ao processar a imagem: {e}")