Update app.py

app.py

@@ -6,7 +6,7 @@ import base64
 
 api_url = "https://5cb20b40-572c-426f-9466-995256f9b6eb.id.repl.co/generate_image"
 
-def generate_image(model, prompt, seed, negative_prompt, sampler, steps):
+def generate_image(model="Abyss OrangeMix", prompt="", seed=0, negative_prompt="", sampler="k_dpmpp_2s_a", steps=50):
     data = "?model=" + model + "&prompt=" + prompt + "&seed=" + str(seed) + "&negative_prompt=" + negative_prompt + "&sampler=" + sampler + "&steps=" + str(steps)
     response = requests.post(api_url + data, timeout=400)
     if response.status_code == 200:
@@ -17,17 +17,18 @@ def generate_image(model, prompt, seed, negative_prompt, sampler, steps):
     else:
         return None
 
-dropdown_model = gr.inputs.Dropdown(['Seek.art MEGA', 'Deliberate', 'Dreamlike Photoreal', 'Realistic Vision'], label="Model", default="Seek.art MEGA")
-textbox_prompt = gr.inputs.Textbox(label="Prompt")
-number_seed = gr.inputs.Number(label="Seed", default=0)
-textbox_negative_prompt = gr.inputs.Textbox(label="Negative Prompt", default="")
-dropdown_sampler = gr.inputs.Dropdown(["k_lms", "k_heun", "k_euler", "k_euler_a", "k_dpm_2", "k_dpm_2_a", "DDIM", "k_dpm_fast", "k_dpm_adaptive", "k_dpmpp_2m", "k_dpmpp_2s_a", "k_dpmpp_sde"], label="Sampler", default="k_dpmpp_2s_a")
-number_steps = gr.inputs.Number(label="Steps", default=50)
-
-inputs = [dropdown_model, textbox_prompt, number_seed, textbox_negative_prompt, dropdown_sampler, number_steps]
+inputs = [
+     gr.inputs.Dropdown(['Abyss OrangeMix', 'AbyssOrangeMix-AfterDark','Anime Pencil Diffusion', 'Anygen', 'Anything Diffusion', 'Anything v3', 'anything_v4_inpainting', 'Arcane Diffusion', 'BPModel', 'Counterfeit', 'Cyberpunk Anime Diffusion', 'CyriousMix', 'DGSpitzer Art Diffusion', 'Dreamshaper', 'DucHaiten Classic Anime',  'Eimis Anime Diffusion', 'Ghibli Diffusion',   'GuoFeng', 'Hentai Diffusion', 'Kenshi', 'Midjourney Diffusion',  'NeverEnding Dream',  'Openniji', 'Pastel Mix', 'Protogen Anime', 'Rev Animated'], label="Model", default="Abyss OrangeMix"),
+    gr.inputs.Textbox(label="Prompt"),
+    gr.inputs.Number(label="Seed", default=0),
+    gr.inputs.Textbox(label="Negative Prompt", default=""),
+    gr.inputs.Dropdown(["k_lms", "k_heun", "k_euler", "k_euler_a", "k_dpm_2", "k_dpm_2_a", "DDIM", "k_dpm_fast", "k_dpm_adaptive", "k_dpmpp_2m", "k_dpmpp_2s_a", "k_dpmpp_sde"], label="Sampler", default="k_dpmpp_2s_a"),
+    gr.inputs.Number(label="Steps", default=50)
+]
 
 outputs = gr.outputs.Image(label="Generated Image", type="pil")
 
+
 interface = gr.Interface(
     fn=generate_image, 
     inputs=inputs, 
@@ -40,3 +41,5 @@ interface = gr.Interface(
 )
 
 interface.launch()
+
+