Update app.py

app.py

@@ -54,14 +54,11 @@ def get_embedding(text: str) -> List[float]:
     
     return embeddings[0].tolist()
 
-@app.post("/v1/embeddings", response_model=EmbeddingResponse)
 @spaces.GPU()
-async def create_embeddings(request: EmbeddingRequest):
-    if isinstance(request.input, str):
-        input_texts = [request.input]
-    else:
-        input_texts = request.input
-
+def process_embeddings(request_dict: dict) -> dict:
+    """非异步函数处理嵌入向量"""
+    input_texts = [request_dict["input"]] if isinstance(request_dict["input"], str) else request_dict["input"]
+    
     embeddings = []
     total_tokens = 0
     
@@ -77,22 +74,30 @@ async def create_embeddings(request: EmbeddingRequest):
             "index": len(embeddings)
         })
 
-    response = EmbeddingResponse(
-        data=embeddings,
-        model=request.model or model_name,
-        usage={
+    return {
+        "object": "list",
+        "data": embeddings,
+        "model": request_dict.get("model", model_name),
+        "usage": {
             "prompt_tokens": total_tokens,
             "total_tokens": total_tokens
         }
-    )
-    
-    return response
+    }
+
+@app.post("/v1/embeddings", response_model=EmbeddingResponse)
+async def create_embeddings(request: EmbeddingRequest):
+    """异步API端点"""
+    result = process_embeddings(request.dict())
+    return result
 
 @spaces.GPU()
 def gradio_embedding(text: str) -> Dict:
-    request = EmbeddingRequest(input=text)
-    response = create_embeddings(request)
-    return response.dict()
+    """Gradio接口函数"""
+    request_dict = {
+        "input": text,
+        "model": model_name
+    }
+    return process_embeddings(request_dict)
 
 # 创建 Gradio 界面
 demo = gr.Interface(