albin commited on
Commit
8f75466
·
1 Parent(s): 29991ef

correct bug access features

Browse files
Files changed (1) hide show
  1. main.py +11 -11
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.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]
 
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]