TheMaisk commited on
Commit
26b64cf
·
verified ·
1 Parent(s): 02a4a22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -21
app.py CHANGED
@@ -6,7 +6,7 @@ import base64
6
 
7
  api_url = "https://5cb20b40-572c-426f-9466-995256f9b6eb.id.repl.co/generate_image"
8
 
9
- def generate_image(model, prompt, seed, negative_prompt, sampler, steps):
10
  data = "?model=" + model + "&prompt=" + prompt + "&seed=" + str(seed) + "&negative_prompt=" + negative_prompt + "&sampler=" + sampler + "&steps=" + str(steps)
11
  response = requests.post(api_url + data, timeout=400)
12
  if response.status_code == 200:
@@ -17,26 +17,21 @@ def generate_image(model, prompt, seed, negative_prompt, sampler, steps):
17
  else:
18
  return None
19
 
20
- dropdown_model = gr.inputs.Dropdown(['Seek.art MEGA', 'Deliberate', 'Dreamlike Photoreal', 'Realistic Vision'], label="Model", default="Seek.art MEGA")
21
- textbox_prompt = gr.inputs.Textbox(label="Prompt")
22
- number_seed = gr.inputs.Number(label="Seed", default=0)
23
- textbox_negative_prompt = gr.inputs.Textbox(label="Negative Prompt", default="")
24
- 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")
25
- number_steps = gr.inputs.Number(label="Steps", default=50)
26
-
27
- inputs = [dropdown_model, textbox_prompt, number_seed, textbox_negative_prompt, dropdown_sampler, number_steps]
28
 
29
  outputs = gr.outputs.Image(label="Generated Image", type="pil")
30
 
31
- interface = gr.Interface(
32
- fn=generate_image,
33
- inputs=inputs,
34
- outputs=outputs,
35
- title="Image Generator",
36
- description="Select options and enter a prompt to generate an image.",
37
- examples=[],
38
- layout="vertical",
39
- theme="ParityError/Interstellar" # Eines der Standard-Themes
40
- )
41
-
42
- interface.launch()
 
6
 
7
  api_url = "https://5cb20b40-572c-426f-9466-995256f9b6eb.id.repl.co/generate_image"
8
 
9
+ def generate_image(model="Abyss OrangeMix", prompt="", seed=0, negative_prompt="", sampler="k_dpmpp_2s_a", steps=50):
10
  data = "?model=" + model + "&prompt=" + prompt + "&seed=" + str(seed) + "&negative_prompt=" + negative_prompt + "&sampler=" + sampler + "&steps=" + str(steps)
11
  response = requests.post(api_url + data, timeout=400)
12
  if response.status_code == 200:
 
17
  else:
18
  return None
19
 
20
+ inputs = [
21
+ 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"),
22
+ gr.inputs.Textbox(label="Prompt"),
23
+ gr.inputs.Number(label="Seed", default=0),
24
+ gr.inputs.Textbox(label="Negative Prompt", default=""),
25
+ 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"),
26
+ gr.inputs.Number(label="Steps", default=50)
27
+ ]
28
 
29
  outputs = gr.outputs.Image(label="Generated Image", type="pil")
30
 
31
+ interface = gr.Interface(generate_image, inputs, outputs, title="",
32
+ description="<center></center>",
33
+ examples=[])
34
+
35
+ interface.launch()
36
+
37
+