update

app.py

@@ -122,9 +122,9 @@ def create_inference_demo(func: inference_fn) -> gr.Blocks:
             with gr.Column():
                 result = gr.Image(label='Result')
 
-        reload_button.click(fn=reload_custom_diffusion_weight_list,
-                            inputs=None,
-                            outputs=weight_name)
+        # reload_button.click(fn=reload_custom_diffusion_weight_list,
+        #                     inputs=None,
+        #                     outputs=weight_name)
         prompt.submit(fn=func,
                       inputs=[
                           model_id,

inference.py

@@ -56,8 +56,8 @@ def inference_fn(
     image_root_folder = os.path.join(model_id, 'inference')
     os.makedirs(image_root_folder, exist_ok = True)
 
-    if prompt is None and args.template_name is None:
-        raise ValueError("please input a single prompt through'--prompt' or select a batch of prompts using '--template_name'.")
+    # if prompt is None and args.template_name is None:
+    #     raise ValueError("please input a single prompt through'--prompt' or select a batch of prompts using '--template_name'.")
 
     # single text prompt
     if prompt is not None:
@@ -65,9 +65,9 @@ def inference_fn(
     else:
         prompt_list = []
 
-    if args.template_name is not None:
-        # read the selected text prompts for generation
-        prompt_list.extend(inference_templates[args.template_name])
+    # if args.template_name is not None:
+    #     # read the selected text prompts for generation
+    #     prompt_list.extend(inference_templates[args.template_name])
 
     for prompt in prompt_list:
         # insert relation prompt <R>
@@ -93,4 +93,4 @@ def inference_fn(
         return image_grid
 
 if __name__ == "__main__":
-    main()
+    inference_fn()