Spaces:
Sleeping
Sleeping
Commit
·
e2f5f01
0
Parent(s):
Duplicate from curtpond/lr_prototype
Browse files- .gitattributes +34 -0
- README.md +14 -0
- app.py +55 -0
- lr_021223.pkl +3 -0
- lr_021823.pkl +3 -0
- requirements.txt +4 -0
- vectorizer_021223.pkl +3 -0
- vectorizer_021823.pkl +3 -0
.gitattributes
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
29 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
30 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
31 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
32 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
title: Lr Prototype
|
3 |
+
emoji: 🦀
|
4 |
+
colorFrom: red
|
5 |
+
colorTo: blue
|
6 |
+
sdk: gradio
|
7 |
+
sdk_version: 3.18.0
|
8 |
+
app_file: app.py
|
9 |
+
pinned: false
|
10 |
+
license: cc
|
11 |
+
duplicated_from: curtpond/lr_prototype
|
12 |
+
---
|
13 |
+
|
14 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Imports
|
2 |
+
import gradio as gr
|
3 |
+
from sklearn.linear_model import LogisticRegression
|
4 |
+
import pickle5 as pickle
|
5 |
+
import re
|
6 |
+
import string
|
7 |
+
import nltk
|
8 |
+
from nltk.corpus import stopwords
|
9 |
+
nltk.download('stopwords')
|
10 |
+
from sklearn.feature_extraction.text import CountVectorizer
|
11 |
+
from sklearn.feature_extraction.text import TfidfVectorizer
|
12 |
+
|
13 |
+
# Load pickled model and vectorizer
|
14 |
+
model = 'lr_021823.pkl'
|
15 |
+
model_loaded = pickle.load(open(model, 'rb'))
|
16 |
+
vectorizer = 'vectorizer_021823.pkl'
|
17 |
+
vectorizer_loaded = pickle.load(open(vectorizer, 'rb'))
|
18 |
+
|
19 |
+
|
20 |
+
# Process input text, including removing stopwords, converting to lowercase, and removing punctuation
|
21 |
+
stop = stopwords.words('english')
|
22 |
+
def process_text(text):
|
23 |
+
text = [word for word in text.split() if word not in stop]
|
24 |
+
text = str(text).lower()
|
25 |
+
text = re.sub(
|
26 |
+
f"[{re.escape(string.punctuation)}]", " ", text
|
27 |
+
)
|
28 |
+
text = " ".join(text.split())
|
29 |
+
return text
|
30 |
+
|
31 |
+
# Vectorize text
|
32 |
+
def vectorize_text(text):
|
33 |
+
text = process_text(text)
|
34 |
+
text = vectorizer_loaded.transform([text])
|
35 |
+
return text
|
36 |
+
|
37 |
+
# Valid input for the model so number of features match
|
38 |
+
def class_predict(text):
|
39 |
+
text = process_text(text)
|
40 |
+
vec = vectorizer_loaded.transform([text])
|
41 |
+
prediction = model_loaded.predict(vec)
|
42 |
+
return prediction
|
43 |
+
|
44 |
+
|
45 |
+
# Define interface
|
46 |
+
demo = gr.Interface(fn=class_predict,
|
47 |
+
title="Text Classification Demo",
|
48 |
+
description="This is a demo of a text classification model using Logistic Regression.",
|
49 |
+
inputs=gr.Textbox(lines=10, placeholder='Input text here...', label="Input Text"),
|
50 |
+
outputs=gr.Textbox(label="Predicted Label: Healthcare: 0, Other: 1, Technology: 2", lines=2, placeholder='Predicted label will appear here...'),
|
51 |
+
allow_flagging='never'
|
52 |
+
)
|
53 |
+
|
54 |
+
demo.launch()
|
55 |
+
|
lr_021223.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f7036cd9deddab77f465e6ab534fdd70552345530e10360e2ffc971c2d628e81
|
3 |
+
size 132
|
lr_021823.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:633ffaf2cdcd72308a63c45058e640e5fde3a97aa07559a112fa616992ec62ef
|
3 |
+
size 3886406
|
requirements.txt
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
scikit-learn==1.1.1
|
2 |
+
pickle5
|
3 |
+
flair
|
4 |
+
nltk==3.8.1
|
vectorizer_021223.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a5c0a94be456efa18f6357794cea11ec493939426b35073aa737de9d31edde7a
|
3 |
+
size 132
|
vectorizer_021823.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:539dadc0945808b81723497ab0a9a5c9ddf2bc0ddf724a946f41e3cbc4846662
|
3 |
+
size 2375371
|