Spaces:
Runtime error
Runtime error
William_S
commited on
Commit
•
1185ef6
1
Parent(s):
e3d8930
grab first result of prediction array
Browse files
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]
|