sachin commited on
Commit
a506b86
·
1 Parent(s): 475b0b9

test-gemma-llm

Browse files
src/server/auth.py CHANGED
@@ -1,7 +1,7 @@
1
  from fastapi.security import APIKeyHeader
2
  from fastapi import HTTPException, status, Depends
3
  from pydantic_settings import BaseSettings
4
- from config.logging_config import logger
5
 
6
  class Settings(BaseSettings):
7
  api_key: str
 
1
  from fastapi.security import APIKeyHeader
2
  from fastapi import HTTPException, status, Depends
3
  from pydantic_settings import BaseSettings
4
+ from logging_config import logger
5
 
6
  class Settings(BaseSettings):
7
  api_key: str
src/server/gemma_llm.py CHANGED
@@ -1,5 +1,5 @@
1
  import torch
2
- from config.logging_config import logger
3
  from transformers import AutoProcessor, Gemma3ForConditionalGeneration
4
  from PIL import Image
5
  from fastapi import HTTPException
 
1
  import torch
2
+ from logging_config import logger
3
  from transformers import AutoProcessor, Gemma3ForConditionalGeneration
4
  from PIL import Image
5
  from fastapi import HTTPException
src/server/logging_config.py CHANGED
@@ -1,7 +1,7 @@
1
  import logging
2
  import logging.config
3
  from logging.handlers import RotatingFileHandler
4
- from .tts_config import config
5
 
6
  logging_config = {
7
  "version": 1,
 
1
  import logging
2
  import logging.config
3
  from logging.handlers import RotatingFileHandler
4
+ from tts_config import config
5
 
6
  logging_config = {
7
  "version": 1,
src/server/main.py CHANGED
@@ -19,7 +19,7 @@ import requests
19
  from logging_config import logger
20
  from tts_config import SPEED, ResponseFormat, config as tts_config
21
  from gemma_llm import LLMManager
22
- from utils.auth import get_api_key, settings as auth_settings
23
 
24
  # Supported language codes
25
  SUPPORTED_LANGUAGES = {
 
19
  from logging_config import logger
20
  from tts_config import SPEED, ResponseFormat, config as tts_config
21
  from gemma_llm import LLMManager
22
+ from auth import get_api_key, settings as auth_settings
23
 
24
  # Supported language codes
25
  SUPPORTED_LANGUAGES = {