Update app.py

app.py

@@ -80,21 +80,13 @@ y_pred = model.predict(x_test)
 
 def get_feats():
   feats = [abs(x) for x in model.coef_[0]]
-  feats, cols = zip(*sorted(zip(feats, data.columns)))
-
-  output = []
-
-  for idx, col in enumerate(reversed(cols)):
-    output.append(col)
-  
-  # max_val = max(feats)
-  # idx = feats.index(max_val)
-  # return data.columns[idx]
-  return output
+  max_val = max(feats)
+  idx = feats.index(max_val)
+  return data.columns[idx]
   
 acc = str(round(metrics.accuracy_score(y_test, y_pred) * 100, 2)) + '%'
-feats = get_feats()
-info = get_article(acc, feats)
+feat = get_feat()
+info = get_article(acc, feat)