pdarleyjr commited on
Commit
2e8b75e
·
1 Parent(s): 9c9cdc6

Fix logging and numpy compatibility issues

Browse files
Files changed (2) hide show
  1. app.py +3 -8
  2. requirements.txt +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from pydantic import BaseModel
5
  from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoConfig
6
  import torch
7
  import os
 
8
  import traceback
9
  from typing import Optional, Dict, Any
10
  from accelerate import Accelerator
@@ -12,16 +13,10 @@ import time
12
  import psutil
13
  from loguru import logger
14
 
15
- # Configure production logging
16
  logger.remove() # Remove default handler
17
  logger.add(
18
- "app.log",
19
- rotation="500 MB",
20
- retention="10 days",
21
- compression="zip",
22
- backtrace=True,
23
- diagnose=True,
24
- enqueue=True,
25
  level="INFO",
26
  format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
27
  )
 
5
  from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoConfig
6
  import torch
7
  import os
8
+ import sys
9
  import traceback
10
  from typing import Optional, Dict, Any
11
  from accelerate import Accelerator
 
13
  import psutil
14
  from loguru import logger
15
 
16
+ # Configure production logging to stderr
17
  logger.remove() # Remove default handler
18
  logger.add(
19
+ sys.stderr,
 
 
 
 
 
 
20
  level="INFO",
21
  format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
22
  )
requirements.txt CHANGED
@@ -5,6 +5,7 @@ uvicorn[standard]==0.27.0
5
  # ML Dependencies
6
  transformers==4.36.2
7
  torch==2.1.2
 
8
  safetensors==0.4.1
9
  accelerate==0.25.0
10
  sentencepiece==0.1.99
 
5
  # ML Dependencies
6
  transformers==4.36.2
7
  torch==2.1.2
8
+ numpy<2.0.0 # Pin numpy to 1.x for compatibility
9
  safetensors==0.4.1
10
  accelerate==0.25.0
11
  sentencepiece==0.1.99