Maybe now?
This commit is contained in:
9
api.py
9
api.py
@@ -3,6 +3,7 @@ import sys
|
||||
|
||||
import uvicorn
|
||||
from fastapi import Body, FastAPI
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.responses import FileResponse
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from pydantic import BaseModel
|
||||
@@ -35,6 +36,14 @@ class PredictResponse(BaseModel):
|
||||
# --- FastAPI App ---
|
||||
app = FastAPI()
|
||||
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"],
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
|
||||
def get_model_for_n(n: int):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user