Update app.py

app.py

@@ -21,7 +21,7 @@ prompt_prefixes = {
 
 current_model = models[0]
 
-#auth_token = os.environ.get("adminToken") or True
+#auth_token = os.environ.get("test") or True
 #pipe = StableDiffusionPipeline.from_pretrained(current_model, use_auth_token=auth_token, torch_dtype=torchfloat, revision="fp16")
 pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torchfloat, revision="fp16")
 
@@ -44,9 +44,10 @@ def on_model_change(model):
 
 def inference(prompt, guidance, steps):
 
+    promptPrev = prompt
     prompt = prompt_prefixes[current_model] + prompt
     image = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512).images[0]
-    return image, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    return image, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(placeholder=promptPrev)
 
 def inference_example(prompt, guidance, steps):
 
@@ -131,11 +132,10 @@ with gr.Blocks(css=css) as demo:
           share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
           
     model.change(on_model_change, inputs=model, outputs=[])
-    run.click(inference, inputs=[prompt, guidance, steps], outputs=[image_out, share_button, community_icon, loading_icon])
+    run.click(inference, inputs=[prompt, guidance, steps], outputs=[image_out, share_button, community_icon, loading_icon, prompt])
     
     share_button.click(None, [], [], _js=share_js)
     
-     
     gr.Examples([
         ["portrait of anime girl", 7.5, 27],
         ["a beautiful perfect face girl, Anime fine details portrait of school girl in front of modern tokyo city landscape on the background deep bokeh, anime masterpiece by studio ghibli, 8k, sharp high quality anime, artstation", 7.5, 27],