Nightfury16 commited on
Commit
2e13826
·
1 Parent(s): 499c62b

Initial commit: Dockerized SVM classifier with FastAPI and Gradio UI.

Browse files
READMe.md CHANGED
@@ -4,7 +4,7 @@ sdk: docker
4
  app_port: 7860
5
  colorFrom: red
6
  colorTo: indigo
7
- description: Low-latency STOP/NOT_STOP text classification using Linear SVM deployed with FastAPI and Docker.
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 = "checkpoint"
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