Spaces:
Running
Running
File size: 1,334 Bytes
17f3a9b d794995 d4b1508 17f3a9b 2ae9fb3 17f3a9b 2ae9fb3 17f3a9b d4b1508 17f3a9b 2ae9fb3 17f3a9b 8d84024 d4b1508 efad2c7 a65e7e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import os
from fastapi import FastAPI
from pydantic import BaseModel
from fastapi.middleware.wsgi import WSGIMiddleware
from transformers import pipeline
app = FastAPI() # 创建 FastAPI 应用
# 定义请求模型
class TextRequest(BaseModel):
text: str
# 定义两个 API 路由处理函数
@app.post("/api/aaa")
async def api_aaa_post(request: TextRequest):
result = request.text + 'aaa'
return {"result": result}
# 定义两个 API 路由处理函数
@app.post("/aaa")
async def aaa(request: TextRequest):
result = request.text + 'aaa'
return {"result": result}
# 定义两个 API 路由处理函数
@app.get("/aaa")
async def api_aaa_get(request: TextRequest):
result = request.text + 'aaa'
return {"result": result}
@app.post("/api/bbb")
async def api_bbb(request: TextRequest):
result = request.text + 'bbb'
return {"result": result}
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
@app.get("/infer_t5")
def t5_get(input):
output = pipe_flan(input)
return {"output": output[0]["generated_text"]}
@app.post("/infer_t5")
def t5_post(input):
output = pipe_flan(input)
return {"output": output[0]["generated_text"]}
@app.get("/")
async def root():
return {"message": "Welcome to the API. Use /api/aaa or /api/bbb for processing."}
|