diff --git a/predict.py b/predict.py
index 6929115..bd9e818 100644
--- a/predict.py
+++ b/predict.py
@@ -15,4 +15,4 @@ with open(f'{PATH}{input}', 'r') as file:
     json_data = json.load(file)
 
     print(f"Input: {json_data['inputs']}")
-    print(f"Prediction: {model.predict(np.array([json_data['inputs']], dtype=np.float32))}")
\ No newline at end of file
+    print(f"Prediction: {model.predict(np.array([float(json_data['inputs'])]))}")
\ No newline at end of file