From 9ed038557c9a6df5acc2027e3d5840972f4056ae Mon Sep 17 00:00:00 2001 From: itqop Date: Mon, 3 Mar 2025 04:23:04 +0300 Subject: [PATCH] fix --- summarize_service/app/inference_service.py | 2 +- summarize_service/app/redis_client.py | 2 -- summarize_service/app/worker.py | 6 +++--- 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/summarize_service/app/inference_service.py b/summarize_service/app/inference_service.py index 800c2e0..4da6d27 100644 --- a/summarize_service/app/inference_service.py +++ b/summarize_service/app/inference_service.py @@ -1,4 +1,4 @@ -from app.config import MAX_INPUT_LENGTH, MAX_OUTPUT_LENGTH +from config import MAX_INPUT_LENGTH, MAX_OUTPUT_LENGTH class InferenceService: def __init__(self, model_loader: "ModelLoader"): diff --git a/summarize_service/app/redis_client.py b/summarize_service/app/redis_client.py index 408f7f5..a07cc4c 100644 --- a/summarize_service/app/redis_client.py +++ b/summarize_service/app/redis_client.py @@ -1,7 +1,5 @@ # app/redis_client.py -import os import json -import time import redis from typing import List from pydantic import BaseModel diff --git a/summarize_service/app/worker.py b/summarize_service/app/worker.py index 159a159..29a060a 100644 --- a/summarize_service/app/worker.py +++ b/summarize_service/app/worker.py @@ -1,8 +1,8 @@ # app/worker.py import time -from app.model_loader import ModelLoader -from app.inference_service import InferenceService -from app.redis_client import RedisClient +from model_loader import ModelLoader +from inference_service import InferenceService +from redis_client import RedisClient from config import BASE_MODEL, ADAPTER_DIR, HF_TOKEN, REDIS_HOST, REDIS_PORT, TEXT_RESULT_CHANNEL, TEXT_TASK_CHANNEL, BATCH_SIZE, WAIT_TIMEOUT def main():