William_S commited on
Commit
1185ef6
1 Parent(s): e3d8930

grab first result of prediction array

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def earthquake(latitude, longitude, time):
24
  input_list.append(longitude)
25
  input_list.append(time)
26
  # prediction result of the model
27
- res = model.predict(np.asarray(input_list).reshape(1, -1))
28
  # split the result to 2 separate variables
29
  print(res)
30
  depth = res[0]
 
24
  input_list.append(longitude)
25
  input_list.append(time)
26
  # prediction result of the model
27
+ res = model.predict(np.asarray(input_list).reshape(1, -1))[0]
28
  # split the result to 2 separate variables
29
  print(res)
30
  depth = res[0]