File size: 635 Bytes
8b90ad5 ecd9090 2eb1363 967efaf 0250d76 fac22d0 fce8087 b9c73b1 696db06 01b0e1c f364dca 01b0e1c 58bcd2f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import Linlada
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
app = FastAPI()
app.add_middleware( # add the middleware
CORSMiddleware,
allow_credentials=True, # allow credentials
allow_origins=["*"], # allow all origins
allow_methods=["*"], # allow all methods
allow_headers=["*"], # allow all headers
)
@app.get("/")
def hello():
return "Hello, I'm Artist"
@app.post("/generate/{message}")
def generate_completion(message: str):
response = Linlada._create_completion(model='gpt-4', messages=[{"role": 'user', "content": message}], stream=False)
return response
|