Upload 2 files

app.py

@@ -720,6 +720,14 @@ class GuiSD:
 
 sd_gen = GuiSD()
 
+@spaces.GPU
+def sd_gen_load_new_model(*args):
+    yield from sd_gen.load_new_model(*args)
+
+@spaces.GPU
+def sd_gen_generate_pipeline(*args):
+    yield from sd_gen.generate_pipeline(*args)
+
 ## BEGIN MOD
 CSS ="""
 .gradio-container, #main { width:100%; height:100%; max-width:100%; padding-left:0; padding-right:0; margin-left:0; margin-right:0; !important; }
@@ -1415,7 +1423,8 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
         translate_prompt_button.click(translate_prompt, [series_dbt], [series_dbt], queue=False)
 
         generate_button.click(
-            fn=sd_gen.load_new_model,
+            #fn=sd_gen.load_new_model,
+            fn=sd_gen_load_new_model,
             inputs=[
                 model_name_gui,
                 vae_model_gui,
@@ -1425,7 +1434,8 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
             queue=True,
             show_progress="minimal",
         ).success(
-            fn=sd_gen.generate_pipeline,
+            #fn=sd_gen.generate_pipeline,
+            fn=sd_gen_generate_pipeline,
             inputs=[
                 prompt_gui,
                 neg_prompt_gui,