Update main.py

main.py

@@ -4,7 +4,6 @@ import gradio as gr
 access_token = os.environ.get('HF_TOKEN',None)
 
 with gr.Blocks() as demo:
-    # gr.load("ai-chatbot-degen/stable_diffusion_multi_test_2", hf_token=access_token, src="spaces")
-    gr.load("ai-chatbot-degen/gradio_ui_custom", hf_token=access_token, src="spaces")
+    gr.load("sun000090/gradio_ui_test", hf_token=access_token, src="spaces")
 
 demo.launch()