🐳 add docker file

2025-01-23 15:24:59 +08:00
parent c6af9a0461
commit fc95d73adb
27 changed files with 5117 additions and 269 deletions
--- a/app/auth/api/internal/logic/storage/upload_file_logic.go
+++ b/app/auth/api/internal/logic/storage/upload_file_logic.go
@@ -69,6 +69,13 @@ func (l *UploadFileLogic) UploadFile(r *http.Request) (resp string, err error) {
 	if face != nil {
 		faceId = face.GetFaceId()
 	}
+	// 分类
+	classification, err := l.svcCtx.AiSvcRpc.TfClassification(l.ctx, &pb.TfClassificationRequest{Image: bytes})
+	if err != nil {
+		return "", err
+	}
+	fmt.Println(classification.ClassName)
+	fmt.Println(classification.Score)

 	// 解析 EXIF 信息
 	exif, err := l.parseExifData(result.Exif)