Update app.py

app.py

@@ -34,9 +34,10 @@ if torch.cuda.is_available():
 @spaces.GPU()
 def generate_image(prompt, ckpt="4-Step"):
     global loaded
-    print(prompt, ckpt)
+    
+    prompt = translator.translate(prompt, 'English')
 
-    print(translator.translate(prompt, 'English'))
+    print(prompt)
     
     checkpoint = checkpoints[ckpt][0]
     num_inference_steps = checkpoints[ckpt][1]
@@ -70,10 +71,10 @@ examples = [
 
 with gr.Blocks(css=CSS, theme="soft") as demo:
     gr.HTML("<h1><center>Adobe DMD2🦖</center></h1>")
-    gr.HTML("<p><center><a href='https://huggingface.co/tianweiy/DMD2'>DMD2</a> text-to-image generation</center></p>")
+    gr.HTML("<p><center><a href='https://huggingface.co/tianweiy/DMD2'>DMD2</a> text-to-image generation</center><br><center>Multi-Languages, 4-step is higher quality & 2X slower</center></p>")
     with gr.Group():
         with gr.Row():
-            prompt = gr.Textbox(label='Enter your prompt (English)', scale=8)
+            prompt = gr.Textbox(label='Enter Your Prompt', scale=8)
             ckpt = gr.Dropdown(label='Steps',choices=['1-Step', '4-Step'], value='4-Step', interactive=True)
             submit = gr.Button(scale=1, variant='primary')
     img = gr.Image(label='DMD2 Generated Image')