Update app.py

app.py

@@ -31,7 +31,4 @@ def predict(input : InputData):
     input.input5, input.input6,
     input.input7]])
     prediction = model.predict(data).tolist()
-    return {'prediction' : f'prediction'}
-
-if __name__ == '__main__':
-    app.run(host = '0.0.0.0', port = 5000)
+    return {'prediction' : f'prediction[0]'}