丁云鹏 4 mesi fa
parent
commit
f1442990e3

+ 3 - 2
recommend-model-produce/src/main/python/tools/inferv2.py

@@ -64,8 +64,9 @@ def thread_task(name, file_list, model_file, params_file):
     results = []
     for file_path in file_list:
         results.extend(process_file(file_path, model_file, params_file))
-    output_file = f"/app/data_{os.path.basename(file_list[0])}.json"
-    write_results(results, output_file)
+        output_file = f"/app/data_{os.path.basename(file_path)}.json"
+        write_results(results, output_file)
+    
     print(f"Thread {name}: finishing")
 
 def main():