main.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. from concurrent.futures import ThreadPoolExecutor
  2. import torch
  3. import torch.nn as nn
  4. import torch.nn.functional as F
  5. import uvicorn
  6. from fastapi import FastAPI
  7. from fastapi.middleware.cors import CORSMiddleware
  8. from loguru import logger
  9. import os
  10. import sys
  11. from routers import get_related_words
  12. from routers import script_config
  13. from routers import tengxunyun_server
  14. from script_score.evaluate_script import router as evaluate_script_router
  15. from script_score.lstm_network import SentimentNet
  16. logger.remove() # 删去 import logger之后自动产生的handler,不删除的话会出现重复输出的现象
  17. logger.add("/data/pythonProject/video_to_word/logs/main_server.{time:YYYY-MM-DD}.log",
  18. rotation="00:00",
  19. format="{time:YYYY-MM-DD HH:mm:ss,SSS} [{process}] [{thread}] {level} {file} {line} - {message}",
  20. level="INFO")
  21. threadPool = ThreadPoolExecutor(max_workers=4)
  22. app = FastAPI()
  23. origins = [
  24. "http://192.168.0.195:9001",
  25. "http://192.168.0.195:31013",
  26. "http://192.168.6.220:3000",
  27. "http://192.168.1.72:3000",
  28. "http://192.168.1.91:3000",
  29. "http://192.168.1.34",
  30. "http://192.168.1.34:8000",
  31. "http://192.168.1.105",
  32. "http://192.168.1.105:3000",
  33. "http://111.206.86.186",
  34. "http://111.206.86.186:3000",
  35. "http://adsp.tjyourong.com.cn",
  36. "http://adsp.tjyourong.com.cn:3000",
  37. "http://adsp.c-top.com.cn",
  38. "http://adsp.c-top.com.cn:3000"
  39. ]
  40. app.add_middleware(
  41. CORSMiddleware,
  42. allow_origins=origins,
  43. allow_credentials=True,
  44. allow_methods=["*"],
  45. allow_headers=["*"],
  46. )
  47. app.include_router(get_related_words.router)
  48. app.include_router(tengxunyun_server.router)
  49. app.include_router(script_config.router)
  50. app.include_router(evaluate_script_router)
  51. if __name__ == '__main__':
  52. uvicorn.run(app='main:app', host="0.0.0.0", port=31013, reload=True, debug=True)
  53. # gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令