Update app.py
Browse files
app.py
CHANGED
@@ -69,9 +69,9 @@ def preprocess_input(input_text, tokenizer, max_length):
|
|
69 |
def get_prediction(input_text, input_type):
|
70 |
is_url = input_type == "URL"
|
71 |
if is_url:
|
72 |
-
input_data = preprocess_input(input_text, url_tokenizer, preprocessing_params['
|
73 |
else:
|
74 |
-
input_data = preprocess_input(input_text, html_tokenizer, preprocessing_params['
|
75 |
|
76 |
prediction = model.predict([input_data, input_data])[0][0]
|
77 |
return prediction
|
|
|
69 |
def get_prediction(input_text, input_type):
|
70 |
is_url = input_type == "URL"
|
71 |
if is_url:
|
72 |
+
input_data = preprocess_input(input_text, url_tokenizer, preprocessing_params['max_url_length'])
|
73 |
else:
|
74 |
+
input_data = preprocess_input(input_text, html_tokenizer, preprocessing_params['max_html_length'])
|
75 |
|
76 |
prediction = model.predict([input_data, input_data])[0][0]
|
77 |
return prediction
|