FastAPI_Start / main.py
michaelj's picture
update predict
8546ebb
# https://medium.com/@qacheampong/building-and-deploying-a-fastapi-app-with-hugging-face-9210e9b4a713
# https://huggingface.co/spaces/Queensly/FastAPI_in_Docker
from fastapi import FastAPI
import uvicorn
app = FastAPI()
#Endpoints
#Root endpoints
@app.get("/")
def root():
return {"API": "Sum of 2 Squares"}
@app.post('/predict')
async def predict_with_prompt(prompt: str):
return "你好"+prompt