venv train.json dataset final_model final_model.tflite