Update sft_fastapi.py
Browse files- sft_fastapi.py +3 -0
sft_fastapi.py
CHANGED
|
@@ -7,6 +7,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 7 |
import random
|
| 8 |
import os
|
| 9 |
from typing import List, Optional
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
app = FastAPI(title="PVS Step Recommender API", version="1.0.0")
|
| 12 |
|
|
|
|
| 7 |
import random
|
| 8 |
import os
|
| 9 |
from typing import List, Optional
|
| 10 |
+
import yaml
|
| 11 |
+
from types import SimpleNamespace
|
| 12 |
+
|
| 13 |
|
| 14 |
app = FastAPI(title="PVS Step Recommender API", version="1.0.0")
|
| 15 |
|