sunxy 11 ay önce
ebeveyn
işleme
f619283243
2 değiştirilmiş dosya ile 2 ekleme ve 4 silme
  1. 1 1
      ReadXlsxFile.py
  2. 1 3
      ai_tag_task.py

+ 1 - 1
ReadXlsxFile.py

@@ -4,7 +4,7 @@ import pandas as pd
 
 
 def getVideoInfoInXlxs(xlsx_file):
-    df = pd.read_excel(xlsx_file)
+    df = pd.read_excel(xlsx_file, engine='openpyxl')
 
     feature_data = []
     for index, row in df.iterrows():

+ 1 - 3
ai_tag_task.py

@@ -5,8 +5,6 @@ import datetime
 import time
 import traceback
 import requests
-import multiprocessing
-import ODPSQueryUtil
 from threading import Timer
 from utils import data_check, get_feature_data, asr_validity_discrimination
 from whisper_asr import get_whisper_asr
@@ -279,7 +277,7 @@ def timer_check():
 
 if __name__ == '__main__':
     # timer_check()
-    feature_df = getVideoInfoInXlxs('past_videos.xlsx')
+    feature_df = getVideoInfoInXlxs('aigc-test/past_videos.xlsx')
     video_id_list = feature_df['videoid'].to_list()
     video_info = {}
     for video_id in video_id_list: