Max Bushuev
commited on
Commit
•
9a4ae09
1
Parent(s):
5d743e0
dfdf
Browse files- api/prediction.py +3 -3
api/prediction.py
CHANGED
|
@@ -10,9 +10,9 @@ router = APIRouter(
|
|
| 10 |
)
|
| 11 |
|
| 12 |
|
| 13 |
-
nlp_task = os.getenv('NLP_TASK')
|
| 14 |
-
model_name = os.getenv('MODEL')
|
| 15 |
-
model = pipeline(
|
| 16 |
|
| 17 |
|
| 18 |
@router.post("/get_prediction", name='Получение предсказания модели')
|
|
|
|
| 10 |
)
|
| 11 |
|
| 12 |
|
| 13 |
+
# nlp_task = os.getenv('NLP_TASK')
|
| 14 |
+
# model_name = os.getenv('MODEL')
|
| 15 |
+
model = pipeline("text-classification", model="Maxim01/intent-classification")
|
| 16 |
|
| 17 |
|
| 18 |
@router.post("/get_prediction", name='Получение предсказания модели')
|