Arroz

README.md

@@ -1,5 +1,5 @@
 ---
-title: "Recomendação de API"
+title: "Api for access AI recommendation"
 emoji: "🚀"
 colorFrom: "blue"
 colorTo: "green"

src/1/routes/ai_routes.py

@@ -35,4 +35,4 @@ async def ai(request: Request):
         outputs = model(input_ids, attention_mask=attention_mask)
         logits = outputs.logits
 
-    return logits.tolist()
+    return {"similarity": logits.tolist()}