update: api (#81)

Vincentqyw · Oct 5, 2024 · 13037f0 · 13037f0
1 parent 5a1b91c
commit 13037f0
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 4 deletions.
diff --git a/api/__init__.py b/api/__init__.py
diff --git a/api/client.py b/api/client.py
@@ -116,13 +116,13 @@ def send_generate_request2(image_path: str) -> Dict[str, np.ndarray]:
         "--image0",
         required=False,
         help="Path for the file's melody",
-        default="../datasets/sacre_coeur/mapping_rot/02928139_3448003521_rot45.jpg",
+        default="datasets/sacre_coeur/mapping_rot/02928139_3448003521_rot45.jpg",
     )
     parser.add_argument(
         "--image1",
         required=False,
         help="Path for the file's melody",
-        default="../datasets/sacre_coeur/mapping_rot/02928139_3448003521_rot90.jpg",
+        default="datasets/sacre_coeur/mapping_rot/02928139_3448003521_rot90.jpg",
     )
     args = parser.parse_args()
     for i in range(10):

diff --git a/package.json b/package.json
@@ -0,0 +1,5 @@
+{
+    "engines": {
+        "node": "18.x"
+    }
+}
diff --git a/api/server.py → server.py b/api/server.py → server.py
@@ -1,5 +1,4 @@
 # server.py
-import sys
 from pathlib import Path
 from typing import Union
 
@@ -9,7 +8,6 @@
 from fastapi.responses import JSONResponse
 from PIL import Image
 
-sys.path.append("..")
 from pydantic import BaseModel
 
 from ui.api import ImageMatchingAPI

diff --git a/vercel.json b/vercel.json
@@ -0,0 +1,14 @@
+{
+    "builds": [
+      {
+        "src": "server.py",
+        "use": "@vercel/python"
+      }
+    ],
+    "routes": [
+      {
+        "src": "/(.*)",
+        "dest": "server.py"
+      }
+    ]
+  }