Update app.py

app.py

@@ -32,7 +32,7 @@ def extract_features(password):
 # Function to predict password strength
 def predict_password_strength(password):
     if not model or not vectorizer:
-        return ["", "", "Model or vectorizer not loaded correctly"]
+        return [["", "", "Model or vectorizer not loaded correctly"]]
 
     try:
         # Extract features from the input password
@@ -57,9 +57,10 @@ def predict_password_strength(password):
         else:
             text = "Unknown strength level."
 
-        return [password, prediction, text]
+        # Return the result as a list of lists to match the DataFrame format
+        return [[password, prediction, text]]
     except Exception as e:
-        return [password, "", f"Error during prediction: {e}"]
+        return [[password, "", f"Error during prediction: {e}"]]
 
 # Gradio Interface
 demo = gr.Interface(