Update app.py

app.py

@@ -69,9 +69,9 @@ def preprocess_input(input_text, tokenizer, max_length):
 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_new_url_length'])
+        input_data = preprocess_input(input_text, url_tokenizer, preprocessing_params['max_url_length'])
     else:
-        input_data = preprocess_input(input_text, html_tokenizer, preprocessing_params['max_new_html_length'])
+        input_data = preprocess_input(input_text, html_tokenizer, preprocessing_params['max_html_length'])
     
     prediction = model.predict([input_data, input_data])[0][0]
     return prediction