Upload t2i_space.py

t2i_space.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+import os
 
 
 def is_repo_name(s):
@@ -33,6 +34,7 @@ def save_space_contents(repo_id: str, dir: str):
     if not is_repo_name(repo_id) or not is_repo_exists(repo_id) or not is_repo_t2i(repo_id):
         print(f"Error: Invalid repo ID: {repo_id}. ")
         return []
+    os.makedirs(dir, exist_ok=True)
     model_name = repo_id.split("/")[-1]
     app_py = f"""
 import gradio as gr