Upload app.py

app.py

@@ -55,40 +55,41 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css="", delete_cac
                 recom_pony = gr.Textbox(label="Pony reccomended prompt", value="Pony", visible=False)
         generate_image_btn = gr.Button(value="GENERATE IMAGE", size="lg", variant="primary")
         result_image = gr.Gallery(label="Generated images", columns=1, object_fit="contain", container=True, preview=True, show_label=False, show_share_button=False, show_download_button=True, interactive=False, visible=True, format="png")
-        gr.LoginButton()
+    gr.LoginButton()
+    gr.DuplicateButton(value="Duplicate Space for private use. (This demo does not work on CPU. Requires GPU Space.)")
 
-        gr.on(
-            triggers=[chat_msg.submit, chat_submit.click],
-            fn=dolphin_respond,
-            inputs=[chat_msg, chatbot, chat_model, chat_sysmsg, chat_tokens, chat_temperature, chat_topp, chat_topk, chat_rp],
-            outputs=[chatbot],
-            queue=True,
-            show_progress="full",
-            trigger_mode="once",
-        ).success(dolphin_parse, [chatbot], [output_text, copy_btn, copy_btn_pony]).success(
-            convert_danbooru_to_e621_prompt, [output_text, tag_type], [output_text_pony], queue=False,
-        ).success(insert_recom_prompt, [output_text, dummy_np, recom_animagine], [output_text, dummy_np], queue=False,
-        ).success(insert_recom_prompt, [output_text_pony, dummy_np_pony, recom_pony], [output_text_pony, dummy_np_pony], queue=False)
-        chat_clear.click(lambda: None, None, chatbot, queue=False)
-        chat_model.change(select_dolphin_model, [chat_model], [chat_model, chat_format, chat_model_info], queue=True, show_progress="full")\
-        .success(lambda: None, None, chatbot, queue=False)
-        chat_format.change(select_dolphin_format, [chat_format], [chat_format], queue=False)\
-        .success(lambda: None, None, chatbot, queue=False)
-        chat_mode.change(select_dolphin_sysprompt, [chat_mode], [chat_sysmsg], queue=False)
-        chat_lang.change(select_dolphin_language, [chat_lang], [chat_sysmsg], queue=False)
-        gr.on(
-            triggers=[chat_add_text.submit, chat_add_submit.click],
-            fn=add_dolphin_models,
-            inputs=[chat_add_text, chat_add_format],
-            outputs=[chat_model],
-            queue=False,
-            trigger_mode="once",
-        )
+    gr.on(
+        triggers=[chat_msg.submit, chat_submit.click],
+        fn=dolphin_respond,
+        inputs=[chat_msg, chatbot, chat_model, chat_sysmsg, chat_tokens, chat_temperature, chat_topp, chat_topk, chat_rp],
+        outputs=[chatbot],
+        queue=True,
+        show_progress="full",
+        trigger_mode="once",
+    ).success(dolphin_parse, [chatbot], [output_text, copy_btn, copy_btn_pony]).success(
+        convert_danbooru_to_e621_prompt, [output_text, tag_type], [output_text_pony], queue=False,
+    ).success(insert_recom_prompt, [output_text, dummy_np, recom_animagine], [output_text, dummy_np], queue=False,
+    ).success(insert_recom_prompt, [output_text_pony, dummy_np_pony, recom_pony], [output_text_pony, dummy_np_pony], queue=False)
+    chat_clear.click(lambda: None, None, chatbot, queue=False)
+    chat_model.change(select_dolphin_model, [chat_model], [chat_model, chat_format, chat_model_info], queue=True, show_progress="full")\
+    .success(lambda: None, None, chatbot, queue=False)
+    chat_format.change(select_dolphin_format, [chat_format], [chat_format], queue=False)\
+    .success(lambda: None, None, chatbot, queue=False)
+    chat_mode.change(select_dolphin_sysprompt, [chat_mode], [chat_sysmsg], queue=False)
+    chat_lang.change(select_dolphin_language, [chat_lang], [chat_sysmsg], queue=False)
+    gr.on(
+        triggers=[chat_add_text.submit, chat_add_submit.click],
+        fn=add_dolphin_models,
+        inputs=[chat_add_text, chat_add_format],
+        outputs=[chat_model],
+        queue=False,
+        trigger_mode="once",
+    )
 
-        copy_btn.click(gradio_copy_text, [output_text], js=COPY_ACTION_JS)
-        copy_btn_pony.click(gradio_copy_text, [output_text_pony], js=COPY_ACTION_JS)
+    copy_btn.click(gradio_copy_text, [output_text], js=COPY_ACTION_JS)
+    copy_btn_pony.click(gradio_copy_text, [output_text_pony], js=COPY_ACTION_JS)
 
-        generate_image_btn.click(generate_image, [output_text, dummy_np], [result_image], show_progress="full")
+    generate_image_btn.click(generate_image, [output_text, dummy_np], [result_image], show_progress="full")
 
 
 if __name__ == "__main__":