llm / app.py
ngd1210's picture
Add text-to-image route to the application
5b01a80
raw
history blame
288 Bytes
from huggingface_hub import InferenceClient
from fastapi import FastAPI
from routes import chatCompletion, textToImage
app = FastAPI()
@app.get("/")
async def root():
return {"message": "Hello World"}
app.include_router(chatCompletion.router)
app.include_router(textToImage.router)