github-actions commited on
Commit
13a5160
·
1 Parent(s): 897a34e

Update model artifacts

Browse files
Files changed (4) hide show
  1. app.py +11 -12
  2. requirements.txt +8 -0
  3. saved_model.joblib +3 -0
  4. vectorizer.joblib +3 -0
app.py CHANGED
@@ -1,23 +1,22 @@
1
- # app.py
2
  import gradio as gr
3
  import joblib
 
4
 
5
- # load model & vectorizer
6
- model = joblib.load("saved_model.joblib")
7
- vectorizer = joblib.load("vectorizer.joblib")
8
 
9
  def predict(text):
10
- X = vectorizer.transform([text])
11
- pred = model.predict(X)[0]
12
- return f"Prediksi: {pred}"
 
13
 
14
- iface = gr.Interface(
15
  fn=predict,
16
- inputs=gr.Textbox(lines=3, placeholder="Tulis komentar di sini..."),
17
  outputs="text",
18
- title="Judi Comment Detector",
19
- description="Deteksi apakah komentar mengandung indikasi judi online."
20
  )
21
 
22
  if __name__ == "__main__":
23
- iface.launch()
 
 
1
  import gradio as gr
2
  import joblib
3
+ from src.preprocessing import clean_text
4
 
5
+ model = joblib.load("model/saved_model.joblib")
6
+ vectorizer = joblib.load("model/vectorizer.joblib")
 
7
 
8
  def predict(text):
9
+ clean = clean_text(text)
10
+ vec = vectorizer.transform([clean])
11
+ pred = model.predict(vec)[0]
12
+ return "🚫 Judi Online" if pred == 1 else "✅ Aman"
13
 
14
+ demo = gr.Interface(
15
  fn=predict,
16
+ inputs=gr.Textbox(lines=3, placeholder="Masukkan komentar..."),
17
  outputs="text",
18
+ title="Deteksi Komentar Judi Online"
 
19
  )
20
 
21
  if __name__ == "__main__":
22
+ demo.launch()
requirements.txt CHANGED
@@ -1,3 +1,11 @@
 
 
1
  scikit-learn==1.3.2
2
  joblib==1.3.2
 
 
 
 
3
  gradio==3.50.2
 
 
 
1
+ pandas==2.2.2
2
+ numpy==1.26.4
3
  scikit-learn==1.3.2
4
  joblib==1.3.2
5
+ mlflow==2.7.0
6
+ uvicorn==0.22.0
7
+ python-multipart==0.0.6
8
+ fastapi==0.95.2
9
  gradio==3.50.2
10
+
11
+
saved_model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d2b3207c279e85fedb3b2040e9e5d087fc0d3efc92c5cdb700901de28c1d13a
3
+ size 1339
vectorizer.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9876b26c157992eb2101c1b6207c00905c9c4d6c2d246d3f7d971855626722ed
3
+ size 3719