Quellcode durchsuchen

在readme中修改启动方式:使用原始的bin/python 启动,gunicorn启动方式存在找不到net class的问题,没有找到解决方法

liyuyi@c-top.com.cn vor 3 Jahren
Ursprung
Commit
3f9c22d45f
3 geänderte Dateien mit 16 neuen und 11 gelöschten Zeilen
  1. 7 2
      main.py
  2. 4 1
      readme.md
  3. 5 8
      script_score/evaluate_script.py

+ 7 - 2
main.py

@@ -1,15 +1,20 @@
 from concurrent.futures import ThreadPoolExecutor
-
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
 import uvicorn
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from loguru import logger
+import os
+import sys
 
 from routers import get_related_words
 from routers import script_config
 from routers import tengxunyun_server
-from script_score.lstm_network import SentimentNet
+
 from script_score.evaluate_script import router as evaluate_script_router
+from script_score.lstm_network import SentimentNet
 
 
 logger.remove()  # 删去 import logger之后自动产生的handler,不删除的话会出现重复输出的现象

+ 4 - 1
readme.md

@@ -15,8 +15,11 @@ conda activate video_to_word <br>
 3、拉取最新代码<br>
 git pull http://git.tjyourong.com.cn/liyuyi/video_to_word.git <br>
 
-4、启动服务<br>
+4、启动服务方式一<br>
+方式一存在存在的问题:lstm 的net 找不到,不清楚什么原因,启动方式二没有这个问题。<br>
 gunicorn main:app -b 0.0.0.0:31013 -w 1 -k uvicorn.workers.UvicornWorker <br>
+<br>
+启动服务方式二<br> nohup /data/Miniconda3/envs/video_to_word/bin/python -u /data/pythonProject/video_to_word/main.py >/dev/null 2>&1 &
 
 2、查看日志<br>
 cd /data/pythonProject/video_to_word/logs/<br>

+ 5 - 8
script_score/evaluate_script.py

@@ -28,15 +28,13 @@ from loguru import logger
 from pangres import upsert
 from pydantic import BaseModel, Field
 
+
 curr_path = os.path.abspath(os.path.dirname(__file__))
 project_root_path = curr_path[:curr_path.find("video_to_word") + len("video_to_word")]
 sys.path.append(project_root_path)
 
-# from script_score.lstm_network import SentimentNet
 from config.url_and_db import toutiao_static_video_url, ai_word_engine
 
-random_num = random.random()
-print("OUT random_num", random_num)
 
 with open('/data/pythonProject/video_to_word/script_score/pkl/vocab.pkl', 'rb') as f:
     vocab = pickle.load(f)
@@ -211,6 +209,7 @@ def delete_script_score(item: DeleteScriptScoreRequest):
                    df=add_df,
                    table_name='ctop_ai_script_score',
                    if_row_exists='update')
+        logger.info(f"request body: {item}, response body: {response}")
     except:
         response.code = -1
         response.message = traceback.format_exc()
@@ -226,7 +225,6 @@ def delete_script_score(item: DeleteScriptScoreRequest):
              summary="获取脚本评级列表",
              response_model=GetScriptScoreLstResponse)
 def get_script_score_lst(item: ScriptScoreLstRequest):
-    logger.info(f"request body: {item}")
     response = GetScriptScoreLstResponse(message="查询成功")
     try:
         end_date = item.end_date + timedelta(days=1)
@@ -259,7 +257,6 @@ def get_script_score_lst(item: ScriptScoreLstRequest):
         response.message = traceback.format_exc()
         response.success = False
         logger.error(f"request body: {item}, response body: {response}")
-
     return response
 
 
@@ -301,13 +298,13 @@ def get_script_score(item: GetScriptScoreRequest):
                               con=ai_word_engine,
                               if_exists='append',
                               index=False)
-
-        return response
+        logger.info(f"request body: {item}, response body: {response}")
     except:
         response.code = -1
         response.success = False
         response.message = traceback.format_exc()
-        return response
+        logger.error(f"request body: {item}, response body: {response}")
+    return response
 
 
 if __name__ == '__main__':