Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ from transformers import AutoTokenizer, AutoModel
|
|
| 10 |
from torch.utils.data import Dataset, DataLoader
|
| 11 |
from typing import Optional
|
| 12 |
import uvicorn
|
|
|
|
| 13 |
|
| 14 |
# =========================
|
| 15 |
# CONFIG
|
|
|
|
| 10 |
from torch.utils.data import Dataset, DataLoader
|
| 11 |
from typing import Optional
|
| 12 |
import uvicorn
|
| 13 |
+
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 14 |
|
| 15 |
# =========================
|
| 16 |
# CONFIG
|