Browse Source

上传接口的改动

AnlaAnla 6 days ago
parent
commit
fb7cec86e6
2 changed files with 11 additions and 3 deletions
  1. 3 3
      app/core/config.py
  2. 8 0
      run_PokemonCardSearch.py

+ 3 - 3
app/core/config.py

@@ -10,9 +10,9 @@ class Settings:
     IMAGE_STORAGE_DIR = os.path.join(STATIC_DIR, "images")
     TEMP_UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
 
-    # 模型路径 (请修改为你实际的绝对路径)
-    YOLO_MODEL_PATH = "/path/to/your/yolo.pt"
-    VIT_MODEL_PATH = "/path/to/your/vit/model/folder"
+    # 模型路径
+    YOLO_MODEL_PATH = "/home/martin/ML/Model/card_cls/yolov11n_card_seg01.onnx"
+    VIT_MODEL_PATH = "/home/martin/ML/Model/pokemon_cls/vit-base-patch16-224-Pokemon03"
 
     # Milvus 配置
     MILVUS_HOST = "127.0.0.1"

+ 8 - 0
run_PokemonCardSearch.py

@@ -0,0 +1,8 @@
+import uvicorn
+import socket
+
+if __name__ == "__main__":
+    ipv4 = socket.gethostbyname(socket.gethostname())
+    port = 18082
+    print(f"http://{ipv4}:{port}/docs")
+    uvicorn.run("app.main:app", host="0.0.0.0", port=port)