alfredplpl commited on
Commit
1db878b
·
1 Parent(s): 7893837

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -50,6 +50,13 @@ def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None,
50
  def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui):
51
  # auto prompt correction
52
  cool_japan_type=str(cool_japan_type_ui)
 
 
 
 
 
 
 
53
  prompt=str(prompt_ui)
54
  neg_prompt=str(neg_prompt_ui)
55
  prompt=prompt.lower()
@@ -141,6 +148,7 @@ with gr.Blocks(css=css) as demo:
141
  with gr.Group():
142
  with gr.Row():
143
  cool_japan_type=gr.Radio(["Anime", "Manga", "Game"])
 
144
  cool_japan_type.value="Anime"
145
 
146
  with gr.Row():
 
50
  def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui):
51
  # auto prompt correction
52
  cool_japan_type=str(cool_japan_type_ui)
53
+ if(cool_japan_type=="Manga"):
54
+ cool_japan_type="manga, monochrome, white and black manga"
55
+ elif(cool_japan_type=="Game"):
56
+ cool_japan_type="game"
57
+ else:
58
+ cool_japan_type="anime"
59
+
60
  prompt=str(prompt_ui)
61
  neg_prompt=str(neg_prompt_ui)
62
  prompt=prompt.lower()
 
148
  with gr.Group():
149
  with gr.Row():
150
  cool_japan_type=gr.Radio(["Anime", "Manga", "Game"])
151
+ cool_japan_type.show_label=False
152
  cool_japan_type.value="Anime"
153
 
154
  with gr.Row():