main.py 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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.6.220:3000",
  26. "http://192.168.1.91:3000",
  27. "http://192.168.1.34",
  28. "http://192.168.1.34:8000",
  29. "http://192.168.1.105",
  30. "http://192.168.1.105:3000",
  31. "http://111.206.86.186",
  32. "http://111.206.86.186:3000",
  33. "http://adsp.tjyourong.com.cn",
  34. "http://adsp.tjyourong.com.cn:3000",
  35. "http://adsp.c-top.com.cn",
  36. "http://adsp.c-top.com.cn:3000"
  37. ]
  38. app.add_middleware(
  39. CORSMiddleware,
  40. allow_origins=origins,
  41. allow_credentials=True,
  42. allow_methods=["*"],
  43. allow_headers=["*"],
  44. )
  45. app.include_router(get_related_words.router)
  46. app.include_router(tengxunyun_server.router)
  47. app.include_router(script_config.router)
  48. app.include_router(evaluate_script_router)
  49. if __name__ == '__main__':
  50. uvicorn.run(app='main:app', host="0.0.0.0", port=31013, reload=True, debug=True)
  51. # gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令