Update app.py

app.py

@@ -27,7 +27,7 @@ def remove_punctuation(text):
     punctuation_free = "".join([i for i in text if i not in string.punctuation])
     return punctuation_free
 
-def test_model(text):
+def test_model(text, vectorizer):
     # Convert text to lowercase
     text = text.lower()
 
@@ -54,8 +54,11 @@ def test_model(text):
     # Return the prediction
     return prediction
 
-# Create the Gradio interface
-iface = gr.Interface(fn=test_model, inputs="text", outputs="text")
+# Fit the vectorizer on your training data to build the vocabulary
+train_data = [...]  # Replace [...] with your actual training data
+vectorizer.fit(train_data)
 
-# Launch the interface
+# Create the Gradio interface
+iface = gr.Interface(fn=test_model, inputs=["text", vectorizer], outputs="text", title="Text Classification")
 iface.launch()
+