latest

app.py

@@ -1,30 +1,20 @@
 import gradio as gr
 import joblib
-import re
 
-# Load the model and vectorizer
-model = joblib.load('model.joblib')
-vectorizer = joblib.load('vectorizer.joblib')
-
-# Preprocess function
-def preprocess(text):
-    text = text.lower()
-    text = re.sub(r'\W+', ' ', text)
-    return text
+# Load the pipeline (which includes both the vectorizer and the classifier)
+pipeline = joblib.load('gibberish_detector_pipeline.joblib')
 
 # Prediction function
 def predict(text):
-    text = preprocess(text)
-    X = vectorizer.transform([text])
-    prediction = model.predict(X)[0]
-    return "Gibberish" if prediction == 1 else "Clean"
+    prediction = pipeline.predict([text])[0]
+    return "Gibberish" if prediction == 1 else "Good English"
 
 # Create a Gradio interface
 iface = gr.Interface(fn=predict, 
                      inputs="text", 
                      outputs="text", 
                      title="Gibberish Detector", 
-                     description="Enter a string to check if it's gibberish or clean.")
+                     description="Enter a string to check if it's gibberish or good English.")
 
 # Launch the interface
 iface.launch()