From 183111bd8055a9b6dd626fec5871f3391cbff62a Mon Sep 17 00:00:00 2001 From: Sergey Revyakin Date: Wed, 25 Feb 2026 11:58:55 +0700 Subject: [PATCH] =?UTF-8?q?=D0=B1=D0=B0=D0=B3=20=D1=81=20torchsig?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- NN_server/server.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/NN_server/server.py b/NN_server/server.py index 0611ec7..d5d1c06 100644 --- a/NN_server/server.py +++ b/NN_server/server.py @@ -1,6 +1,8 @@ from flask import Flask, request, jsonify from dotenv import dotenv_values from common.runtime import load_root_env, validate_env, as_int, as_str +import os +import sys import matplotlib.pyplot as plt from Model import Model import numpy as np @@ -12,9 +14,13 @@ import asyncio import shutil import json import gc -import os import logging +TORCHSIG_PATH = "/app/torchsig" +if TORCHSIG_PATH not in sys.path: + # Ensure import torchsig resolves to /app/torchsig/torchsig package. + sys.path.insert(0, TORCHSIG_PATH) + logging.basicConfig(level=logging.INFO) app = Flask(__name__) @@ -115,7 +121,10 @@ def receive_data(): result_msg = {} data_to_send = {} prediction_list = [] + #print(model_list) for model in model_list: + #print(str(freq)) + #print(model.get_model_name()) if str(freq) in model.get_model_name(): print('-' * 100) print(str(model))