完善推理模块的主体功能

app/model/crud/project_detect_crud.py

@@ -171,3 +171,10 @@ def get_log_imgs(log_id: int, session: Session):
     result = [ProjectDetectLogImgOut.from_orm(img) for img in query.all()]
     return result
 
+
+def get_log_img_url(img_id: int, session: Session):
+    result = session.query(ProjectDetectLogImg).filter_by(id=img_id).first()
+    if result is None:
+        return None
+    else:
+        return result.image_url

app/model/schemas/project_detect_schemas.py

@@ -64,7 +64,7 @@ class ProjectDetectLogOut(BaseModel):
     detect_version: Optional[str]
     detect_name: Optional[str]
     train_id: Optional[int]
-    train_version: Optional[int]
+    train_version: Optional[str]
     pt_type: Optional[str]
     create_time: Optional[datetime]