feat: demo support

Author: LeoBorai

README.md

@@ -1,14 +1,14 @@
 ---
 title: amazon-reviews-classification
-emoji: 🐠
-colorFrom: gray
+emoji: 🗣️
+colorFrom: yellow
 colorTo: purple
 sdk: gradio
 sdk_version: 5.20.1
 app_file: app.py
 pinned: false
 license: mit
-short_description: Applying Reviews Classification using Logistic Regression
+short_description: Sentiment Analysis via Logistic Regression
 ---
 
 <div>

app.py

@@ -0,0 +1,19 @@
+import gradio as gr
+import pickle
+
+from sklearn.feature_extraction.text import CountVectorizer
+
+model = pickle.load(open('model.pkl', 'rb'))
+
+def perform(text):
+  vectorizer = CountVectorizer()
+  text_input = vectorizer.transform([text])
+  score = model.predict(text_input)[0]
+
+  return score
+
+demo = gr.Interface(fn=predict,
+  inputs=[gr.Textbox(label="Message")],
+  outputs=gr.Textbox(label="Score"))
+
+demo.launch()