curtpond commited on
Commit
e2f5f01
0 Parent(s):

Duplicate from curtpond/lr_prototype

Browse files
.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