File size: 693 Bytes
11551ca b273357 11551ca b273357 7b36638 c6bff1f 11551ca b273357 7b36638 b273357 7b36638 b273357 7b36638 11551ca 0a323e6 c6bff1f b273357 |
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 |
#!/usr/bin/env python3
import os
import json
import uuid
import uvicorn
import pika
from fastapi import FastAPI, Body, Header, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from worker import main as rabbit_worker
app = FastAPI()
API_KEY = os.getenv("SECRET_KEY")
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/")
async def root():
return {"status": "ok", "message": "API is running"}
if __name__ == "__main__":
os.system('python download_models_hf.py')
rabbit_worker()
uvicorn.run(app, host="0.0.0.0", port=8000) |