Mark-Lasfar
commited on
Commit
·
a94def8
1
Parent(s):
16d7ee8
endpoints.py generation.py
Browse files- api/endpoints.py +2 -0
api/endpoints.py
CHANGED
|
@@ -24,6 +24,8 @@ from utils.constants import MODEL_ALIASES, MODEL_NAME, SECONDARY_MODEL_NAME, TER
|
|
| 24 |
import psutil
|
| 25 |
import time
|
| 26 |
router = APIRouter()
|
|
|
|
|
|
|
| 27 |
logger = logging.getLogger(__name__)
|
| 28 |
|
| 29 |
# Check HF_TOKEN and BACKUP_HF_TOKEN
|
|
|
|
| 24 |
import psutil
|
| 25 |
import time
|
| 26 |
router = APIRouter()
|
| 27 |
+
from pydantic import BaseModel
|
| 28 |
+
|
| 29 |
logger = logging.getLogger(__name__)
|
| 30 |
|
| 31 |
# Check HF_TOKEN and BACKUP_HF_TOKEN
|