Update app.py

app.py

@@ -7,6 +7,9 @@ from diffusers import DiffusionPipeline, AutoencoderTiny
 from huggingface_hub import hf_hub_download
 
 import config
+styles_name = [style["name"] for style in config.style_list]
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
 
 def feifeimodload():
 
@@ -29,12 +32,8 @@ def feifeimodload():
     
 pipe = feifeimodload()
 
-
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 2048
-
 @spaces.GPU()
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, styles_Radio="(None)", seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
@@ -92,7 +91,6 @@ DarkIdol-flux is a text-to-image AI model designed to create aesthetic, detailed
                 value="",
             )
         with gr.Row():
-            styles_name = [style["name"] for style in config.style_list]
             styles_Radio = gr.Dropdown(
                 styles_name,
                 label="Styles",
@@ -155,7 +153,7 @@ DarkIdol-flux is a text-to-image AI model designed to create aesthetic, detailed
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn = infer,
-        inputs = [prompt, seed, randomize_seed, width, height, num_inference_steps],
+        inputs = [prompt, styles_Radio, seed, randomize_seed, width, height, num_inference_steps],
         outputs = [result, seed]
     )