Upload 5 files

README.md

@@ -1,10 +1,10 @@
 ---
-title: Gradio Text-to-Image Demo Space creation helper
+title: Gradio Demo Space creation helper
 emoji: 🐶
 colorFrom: yellow
 colorTo: red
 sdk: gradio
-sdk_version: 4.38.1
+sdk_version: 4.41.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -4,11 +4,11 @@ from t2i_space import get_t2i_space_contents
 css = """"""
 
 with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
-    gr.Markdown("# Gradio Text-to-Image Demo Space creation helper")
+    gr.Markdown("# Gradio Demo Space creation helper")
     gr.Markdown(
         f"""
 **The steps are the following**:
-- Input text-to-image model Repo ID which you want to use. e.g. 'user/model'.
+- Input model Repo ID which you want to use. e.g. 'user/model'.
 - 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.
@@ -18,13 +18,14 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo:
     )
     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)
         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],
+        inputs=[repo_id, gradio_version],
         outputs=[space_file],
     )
 

t2i_space.py

@@ -14,7 +14,7 @@ def is_repo_exists(repo_id):
         if api.repo_exists(repo_id=repo_id, repo_type="model"): 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
 
 
@@ -22,7 +22,7 @@ def is_repo_t2i(repo_id):
     from huggingface_hub import HfApi
     api = HfApi()
     try:
-        model_info =  api.repo_info(repo_id=repo_id, repo_type="model")
+        model_info = api.repo_info(repo_id=repo_id, repo_type="model")
         if model_info.pipeline_tag == "text-to-image": return True
         else: return False
     except Exception as e:
@@ -30,26 +30,25 @@ def is_repo_t2i(repo_id):
         return True # for safe
     
 
-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):
+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}. ")
         return []
     os.makedirs(dir, exist_ok=True)
     model_name = repo_id.split("/")[-1]
     app_py = f"""import gradio as gr
-demo = gr.load("{repo_id}", src="models").launch()
-demo.launch()
+import os
+demo = gr.load("{repo_id}", src="models", hf_token=os.environ.get("HF_TOKEN")).launch()
 """
     readme_md = f"""---
-title: {model_name} Text-to-Image
+title: {model_name} Demo
 emoji: 🖼
 colorFrom: purple
 colorTo: red
 sdk: gradio
-sdk_version: 4.38.1
+sdk_version: {gradio_version}
 app_file: app.py
 pinned: false
-short_description: {repo_id} | Text-to-Image
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
@@ -64,5 +63,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):
-    return save_space_contents(repo_id, "./temp/")
+def get_t2i_space_contents(repo_id: str, gradio_version: str):
+    return save_space_contents(repo_id, gradio_version, "./temp/")

template/README.md

@@ -0,0 +1,12 @@
+---
+title: {model_name} Demo
+emoji: 🖼
+colorFrom: purple
+colorTo: red
+sdk: gradio
+sdk_version: {gradio_version}
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

template/app.py

@@ -0,0 +1,3 @@
+import gradio as gr
+import os
+demo = gr.load("{repo_id}", src="models", hf_token=os.environ.get("HF_TOKEN")).launch()