pushpikaLiyanagama commited on
Commit
545f904
·
verified ·
1 Parent(s): 1af117c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def predict(course_overview, reading_file, abstract_materiale, concrete_material
37
  predictions = {}
38
  for target, model in models.items():
39
  pred = model.predict(input_scaled)
40
- predictions[target] = int(pred[0])
41
 
42
  return predictions
43
 
 
37
  predictions = {}
38
  for target, model in models.items():
39
  pred = model.predict(input_scaled)
40
+ predictions[target] = pred[0] # Removed int() conversion
41
 
42
  return predictions
43