Spaces:
Sleeping
Sleeping
Fix logging and numpy compatibility issues
Browse files- app.py +3 -8
- 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 |
-
|
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
|