Update app.py

app.py

@@ -83,26 +83,24 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=25)
+@spaces.GPU(duration=35)
 def Piper(_do,_dont):
     return pipe(
         _do,
         height=320,
         width=480,
         negative_prompt=_dont,
-        num_inference_steps=50,
-        guidance_scale=2
+        num_inference_steps=100,
+        guidance_scale=4
     )
 
-def infer(prompt,prompt2):
+def infer(prompt_en,prompt2_en):
     name = generate_random_string(12)+".png"
-    prompt_en = translate(prompt,"english")
-    prompt2_en = translate(prompt2,"english")
-    if prompt == None or prompt.strip() == "":
-        _do = 'realistic natural sharp light vivid vintage amazing yet reasonable scene coloring'
+    if prompt_en == "":
+        _do = 'natural sharp light vivid vintage coloring, amazing while also reasonable and realistic scene'
     else:
-        _do = f'realistic natural sharp light vivid vintage { prompt_en } amazing yet reasonable scene coloring'
-    if prompt2 == None or prompt2.strip() == "":
+        _do = f'natural sharp light vivid vintage coloring, amazing while also reasonable and realistic { prompt_en } scene'
+    if prompt2_en == "":
         _dont = 'smooth texture, fictional proportions, blurred content, distorted items, deformed palms, logos and signs, texts and prints'
     else:
         _dont = f'{prompt2_en} where in {prompt_en}, smooth texture, fictional proportions, blurred content, distorted items, deformed palms, logos and signs, texts and prints'
@@ -176,6 +174,8 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Row():
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
+        with gr.Row():
+            result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
     def _ret(idx,p1,p2):
@@ -184,10 +184,12 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         print(f'Finished {idx}: {v}')
         return v
     def _rets(p1,p2):
+        p1_en = translate(p1,"english")
+        p2_en = translate(p2,"english")
         ln = len(result)
         idxs = list(range(ln))
-        p1s = [p1 for _ in idxs]
-        p2s = [p2 for _ in idxs]
+        p1s = [p1_en for _ in idxs]
+        p2s = [p2_en for _ in idxs]
         return list(Pool(ln).imap( _ret, idxs, p1s, p2s ))
     run_button.click(fn=_rets,inputs=[prompt,prompt2],outputs=result)