Update app.py

app.py

@@ -70,10 +70,12 @@ def get_prediction(input_text, input_type):
     is_url = input_type == "URL"
     if is_url:
         input_data = preprocess_input(input_text, url_tokenizer, preprocessing_params['max_url_length'])
+        input_data = [input_data, np.zeros((1, preprocessing_params['max_html_length']))]  # dummy HTML input
     else:
         input_data = preprocess_input(input_text, html_tokenizer, preprocessing_params['max_html_length'])
+        input_data = [np.zeros((1, preprocessing_params['max_url_length'])), input_data]  # dummy URL input
     
-    prediction = model.predict([input_data, input_data])[0][0]
+    prediction = model.predict(input_data)[0][0]
     return prediction
 
 # Gradio UI