Spaces:
Sleeping
Sleeping
Commit
·
2e13826
1
Parent(s):
499c62b
Initial commit: Dockerized SVM classifier with FastAPI and Gradio UI.
Browse files- READMe.md +1 -1
- app.py +1 -1
- checkpoints/._svm_stop_classifier.pkl +3 -0
- checkpoints/._tfidf_vectorizer.pkl +3 -0
- checkpoints/svm_stop_classifier.pkl +3 -0
- checkpoints/tfidf_vectorizer.pkl +3 -0
READMe.md
CHANGED
|
@@ -4,7 +4,7 @@ sdk: docker
|
|
| 4 |
app_port: 7860
|
| 5 |
colorFrom: red
|
| 6 |
colorTo: indigo
|
| 7 |
-
description:
|
| 8 |
---
|
| 9 |
|
| 10 |
# STOP Classifier API
|
|
|
|
| 4 |
app_port: 7860
|
| 5 |
colorFrom: red
|
| 6 |
colorTo: indigo
|
| 7 |
+
description: STOP/NOT_STOP text classification using Linear SVM deployed with FastAPI and Docker.
|
| 8 |
---
|
| 9 |
|
| 10 |
# STOP Classifier API
|
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from fastapi import FastAPI, HTTPException
|
|
| 5 |
from pydantic import BaseModel, Field
|
| 6 |
from typing import List
|
| 7 |
|
| 8 |
-
CHECKPOINT_DIR = "
|
| 9 |
TFIDF_PATH = os.path.join(CHECKPOINT_DIR, "tfidf_vectorizer.pkl")
|
| 10 |
SVM_PATH = os.path.join(CHECKPOINT_DIR, "svm_stop_classifier.pkl")
|
| 11 |
|
|
|
|
| 5 |
from pydantic import BaseModel, Field
|
| 6 |
from typing import List
|
| 7 |
|
| 8 |
+
CHECKPOINT_DIR = "checkpoints"
|
| 9 |
TFIDF_PATH = os.path.join(CHECKPOINT_DIR, "tfidf_vectorizer.pkl")
|
| 10 |
SVM_PATH = os.path.join(CHECKPOINT_DIR, "svm_stop_classifier.pkl")
|
| 11 |
|
checkpoints/._svm_stop_classifier.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:16808147eb96ad4e6f1466d8438da1fb909313e236b918fa4cce15d074c8aa23
|
| 3 |
+
size 4096
|
checkpoints/._tfidf_vectorizer.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:945490cf6a44290ced44147986c07af696aefb18af7a378a3d13af04ffea0532
|
| 3 |
+
size 4096
|
checkpoints/svm_stop_classifier.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fd4c744f7b8886d8ab176b92a11660b877a8790318f54d35c2feba1deace0977
|
| 3 |
+
size 28891
|
checkpoints/tfidf_vectorizer.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b9d0ec1f05ab16b347a8f8fa21f18889c9ceed65af63f25b284bdf3d95860260
|
| 3 |
+
size 36340
|