diff --git a/download_model.py b/download_model.py
new file mode 100644
index 0000000..370e66d
--- /dev/null
+++ b/download_model.py
@@ -0,0 +1,3 @@
+from google.colab import files
+files.download("./OUTPUT_MODEL/G_latest.pth")
+files.download("./OUTPUT_MODEL/config.json")
\ No newline at end of file