Просмотр исходного кода

增加python的typing的版本兼容

AnlaAnla 3 месяцев назад
Родитель
Сommit
3161d184d7
3 измененных файлов с 2 добавлено и 3 удалено
  1. 0 1
      app/api/card_inference.py
  2. 2 1
      app/core/config.py
  3. 0 1
      app/main.py

+ 0 - 1
app/api/card_inference.py

@@ -1,6 +1,5 @@
 from fastapi import APIRouter, File, UploadFile, Depends, HTTPException, Path
 from fastapi.concurrency import run_in_threadpool
-from typing import Annotated
 from enum import Enum
 from ..core.config import settings
 from app.services.card_service import CardInferenceService, card_service

+ 2 - 1
app/core/config.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from typing import Dict
 
 
 # 定义一个模型的配置结构
@@ -16,7 +17,7 @@ class Settings:
 
     # 使用一个字典来管理所有卡片检测模型
     # key (如 'outer_box') 将成为 API 路径中的 {inference_type}
-    CARD_MODELS_CONFIG: dict[str, CardModelConfig] = {
+    CARD_MODELS_CONFIG: Dict[str, CardModelConfig] = {
         "outer_box": {
             "pth_path": "Model/outer_box.pth",
             "class_dict": {1: 'outer_box'},

+ 0 - 1
app/main.py

@@ -1,4 +1,3 @@
-# app/main.py
 from fastapi import FastAPI
 from contextlib import asynccontextmanager
 from .core.model_loader import load_models, unload_models