Spaces:
Sleeping
Sleeping
File size: 796 Bytes
d1cad4b 6558cd8 d1cad4b 6df5942 |
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 |
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from inference import rag_chain
from pydantic import BaseModel
from fastapi.staticfiles import StaticFiles
class Body(BaseModel):
subject: str
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_methods=["*"],
allow_headers=["*"],
)
@app.post("/generate_questions")
async def generate_questions(body: Body):
subject = body.subject
query = f"Quero que você gere questões de biologia, sendo do assunto: {subject}."
res = rag_chain.invoke(f"""{query}""")
return res
app.mount("/", StaticFiles(directory="static", html=True), name="static")
if __name__ == "__main__":
import uvicorn
uvicorn.run("app:app", host="0.0.0.0", port=8000)
|