Upload 2 files

app.py

@@ -12,20 +12,21 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
 - Click "Submit" and download generated README.md and app.py.
 - [Create your new Gradio space](https://huggingface.co/new-space) with blank.
 - Upload README.md and app.py to your space.
-- If you got 500 error, it happens often, just restart space.
+- If you got 500 / 504 error, it happens often, just restart space.
 - If it does not work no matter how many times you try, there is often a problem with the settings of the original repo itself, or there is no generation function.
             """
     )
     with gr.Column():
         repo_id = gr.Textbox(label="Model repo ID", placeholder="username/modelname", value="", max_lines=1)
         gradio_version = gr.Textbox(label="Gradio version", placeholder="username/modelname", value="4.44.0", max_lines=1)
+        private_ok = gr.Checkbox("Allow private repo", value=True)
         run_button = gr.Button(value="Submit")
         space_file = gr.Files(label="Output", interactive=False)
 
     gr.on(
         triggers=[repo_id.submit, run_button.click],
         fn=get_t2i_space_contents,
-        inputs=[repo_id, gradio_version],
+        inputs=[repo_id, gradio_version, private_ok],
         outputs=[space_file],
     )
 

t2i_space.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 import os
-
+import gradio as gr
 
 def is_repo_name(s):
     import re
@@ -26,13 +26,16 @@ def is_repo_t2i(repo_id):
         if model_info.pipeline_tag == "text-to-image": return True
         else: return False
     except Exception as e:
-        print(f"Error: Failed to connect {repo_id}. ")
+        print(f"Error: Failed to connect {repo_id}.")
         return True # for safe
     
 
-def save_space_contents(repo_id: str, gradio_version: 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}. ")
+def save_space_contents(repo_id: str, gradio_version: str, private_ok: bool, dir: str):
+    if not is_repo_name(repo_id): #  or not is_repo_t2i(repo_id)
+        gr.Info(f"Error: Invalid repo ID: {repo_id}.")
+        return []
+    if not private_ok and not is_repo_exists(repo_id):
+        gr.Info(f"Error: Repo doesn't exist: {repo_id}.")
         return []
     os.makedirs(dir, exist_ok=True)
     model_name = repo_id.split("/")[-1]
@@ -63,5 +66,5 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
     return [app_path, readme_path]
 
 
-def get_t2i_space_contents(repo_id: str, gradio_version: str):
-    return save_space_contents(repo_id, gradio_version, "./temp/")
+def get_t2i_space_contents(repo_id: str, gradio_version: str, private_ok: bool):
+    return save_space_contents(repo_id, gradio_version, private_ok, "./temp/")