Update Dockerfile

Dockerfile

@@ -149,6 +149,18 @@ def restore():
 
     if not flag_exists and not force_restore:
         print("Restore skip: initialized.flag not found, this is first deploy or rebuild.")
+        import io
+        flag_content = b"initialized\n"
+        api.upload_file(
+            path_or_fileobj=io.BytesIO(flag_content),
+            path_in_repo=INIT_FLAG,
+            repo_id=repo_id,
+            repo_type="dataset",
+            token=token,
+            commit_message="Create initialized.flag on first deploy",
+        )
+
+        print("initialized.flag created in Dataset.")
         return
 
     if force_restore: