Spaces:
Sleeping
Sleeping
albin
commited on
Commit
·
8f75466
1
Parent(s):
29991ef
correct bug access features
Browse files
main.py
CHANGED
|
@@ -72,17 +72,17 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
|
|
| 72 |
|
| 73 |
features = extract_features(str(url))
|
| 74 |
data = []
|
| 75 |
-
data.append(str(features
|
| 76 |
-
data.extend(int(features
|
| 77 |
-
data.extend(str(features
|
| 78 |
-
data.extend(int(features
|
| 79 |
-
data.extend(str(features
|
| 80 |
-
data.extend(float(features
|
| 81 |
-
data.extend(int(features
|
| 82 |
-
data.extend(int(features
|
| 83 |
-
data.extend(float(features
|
| 84 |
-
data.extend(float(features
|
| 85 |
-
data.extend(int(features
|
| 86 |
|
| 87 |
prediction = model.predict([data])
|
| 88 |
output = prediction[0]
|
|
|
|
| 72 |
|
| 73 |
features = extract_features(str(url))
|
| 74 |
data = []
|
| 75 |
+
data.append(str(features['URL']))
|
| 76 |
+
data.extend(int(features['URLLength']))
|
| 77 |
+
data.extend(str(features['Domain']))
|
| 78 |
+
data.extend(int(features['DomainLength']))
|
| 79 |
+
data.extend(str(features['TLD']))
|
| 80 |
+
data.extend(float(features['CharContinuationRate']))
|
| 81 |
+
data.extend(int(features['TLDLength']))
|
| 82 |
+
data.extend(int(features['NoOfSubDomain']))
|
| 83 |
+
data.extend(float(features['DegitRatioInURL']))
|
| 84 |
+
data.extend(float(features['SpacialCharRatioInURL']))
|
| 85 |
+
data.extend(int(features['IsHTTPS']))
|
| 86 |
|
| 87 |
prediction = model.predict([data])
|
| 88 |
output = prediction[0]
|