Updated numbers of prediction labels.

app.py

@@ -36,11 +36,7 @@ def vectorize_text(text):
     return text
 
 # Valid input for the model so number of features match
-def create_valid_input(text):
-    text = vectorize_text(text)
-    text = text.toarray()
-    text = text.reshape(1, -1)
-    return text
+# Code will go here
 
 # Prediction function
 def predict(text):
@@ -54,7 +50,7 @@ demo = gr.Interface(fn=predict,
                         title="Text Classification Demo",
                         description="This is a demo of a text classification model using Logistic Regression.",
                         inputs=gr.Textbox(lines=10, placeholder='Input text here...', label="Input Text"),
-                        outputs=gr.Textbox(label="Predicted Label: Other: 1, Healthcare: 2, Technology: 3", lines=2, placeholder='Predicted label will appear here...'),
+                        outputs=gr.Textbox(label="Predicted Label: Other: 0, Healthcare: 1, Technology: 2", lines=2, placeholder='Predicted label will appear here...'),
                         allow_flagging='never'
 )