Update app.py

app.py

@@ -36,7 +36,7 @@ def on_token_change(token):
 
     return gr.update(visible=bool(model_names)), gr.update(visible=bool(model_names)), gr.update(choices=model_names), gr.update(visible=bool(model_names)), gr.update(value=error_str(error))
 
-def create_and_push(model_id, title, description, prefix, update, token):
+def create_and_push(space_type, model_id, title, description, prefix, update, token):
 
     try:
 
@@ -50,7 +50,7 @@ def create_and_push(model_id, title, description, prefix, update, token):
         )
 
         # 2. Replace the name, title, and description in the template
-        with open("template/app.py", "r") as f:
+        with open("template/app_simple.py" if space_type == "Simple" else "template/app_advanced.py", "r") as f:
             app = f.read()
         app = app.replace("$model_id", model_id)
         app = app.replace("$title", title)
@@ -71,13 +71,14 @@ def create_and_push(model_id, title, description, prefix, update, token):
         )
 
         # 5. Upload template/requirements.txt to the space
-        api.upload_file(
-            path_or_fileobj="template/requirements.txt",
-            path_in_repo="requirements.txt",
-            repo_id=model_id,
-            token=token,
-            repo_type="space",
-        )
+        if space_type == "Advanced":
+            api.upload_file(
+                path_or_fileobj="template/requirements.txt",
+                path_in_repo="requirements.txt",
+                repo_id=model_id,
+                token=token,
+                repo_type="space",
+            )
 
         # 5. Delete the app.py file
         os.remove("app.py")
@@ -127,6 +128,13 @@ with gr.Blocks() as demo:
                 description = gr.Textbox(label="Description", placeholder="e.g. Demo for my awesome Diffusers model")
                 prefix = gr.Textbox(label="Prefix tokens", placeholder="Tokens that are required to be present in the prompt, e.g. `rick and morty style`")
                 update = gr.Checkbox(label="Update the space if it already exists?")
+
+                # text explainig the difference between simple and advanced types: simple runs on GPU using HF inference API, advanced runs on CPU by default with the option to upgrade to GPU:
+                gr.Markdown("""#### Choose space type
+                - **Simple** - Runs on GPU using Hugging Face inference API, but you cannot control image generation parameters.
+                - **Advanced** - Runs on CPU by default, with the option to upgrade to GPU. You can control image generation parameters: guidance, number of steps, image size, etc. Also supports **image-to-image** generation.""")
+                space_type =gr.Radio(label="Space type", choices=["Simple", "Advanced"], value="Simple")
+
                 brn_create = gr.Button("Create the space")
 
     error_output = gr.Markdown(label="Output")
@@ -148,7 +156,7 @@ with gr.Blocks() as demo:
 
     brn_create.click(
         fn=create_and_push,
-        inputs=[radio_model_names, title, description, prefix, update, input_token],
+        inputs=[space_type, radio_model_names, title, description, prefix, update, input_token],
         outputs=[error_output],
         scroll_to_output=True
     )