Update app.py

app.py

@@ -84,10 +84,11 @@ def get_prediction(input_text):
     url_data = preprocess_input(cleaned_url, url_tokenizer, max_url_length)
     html_data = preprocess_input(cleaned_html, html_tokenizer, max_html_length)
     
-    # Combine URL and HTML data by concatenation
-    combined_data = np.concatenate((url_data, html_data), axis=1)
+    # Ensure the input shapes are correct
+    url_data = np.expand_dims(url_data, axis=0)  # Add batch dimension
+    html_data = np.expand_dims(html_data, axis=0)  # Add batch dimension
     
-    prediction = model.predict(combined_data)[0][0]
+    prediction = model.predict([url_data, html_data])[0][0]
     return prediction
 
 def phishing_detection(input_text):