Upload 3 files

.well-known/openapi.yaml

@@ -0,0 +1,33 @@
+openapi: 3.0.1
+info:
+  title: 實體辨識 API
+  description: 提供 實體辨識 任務分析
+  version: "1.0.0"
+servers:
+  - url: https://turkeyengineer-ckip-ner.hf.space
+paths:
+  /analyze:
+    post:
+      summary: 分析輸入句子
+      operationId: analyze
+      requestBody:
+        required: true
+        content:
+          application/json:
+            schema:
+              type: object
+              properties:
+                sentence:
+                  type: string
+                  example: 這是一個測試句子。
+      responses:
+        '200':
+          description: 成功回傳分析結果
+          content:
+            application/json:
+              schema:
+                type: object
+                properties:
+                  result:
+                    type: string
+                    example: 分析結果

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+from transformers import pipeline, AutoTokenizer
+
+model = pipeline("token-classification", model="ckiplab/bert-base-chinese-ner", tokenizer=AutoTokenizer.from_pretrained("ckiplab/bert-base-chinese-ner"), aggregation_strategy="simple")
+
+def analyze(sentence: str):
+    result = model(sentence)
+    return  
+
+demo = gr.Interface(fn=analyze, inputs="text", outputs="text", title="實體辨識")
+
+app = FastAPI()
+app = gr.mount_gradio_app(app, demo, path="/")
+
+@app.post("/analyze")
+async def api_analyze(request: Request):
+    payload = await request.json()
+    return JSONResponse(content={"result": analyze(payload.get("sentence", ""))})

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+transformers
+torch
+fastapi
+uvicorn