Update app.py

app.py

@@ -44,7 +44,7 @@ height = int(os.getenv("IMAGE_HEIGHT", "512"))
 num_images = int(os.getenv("NUM_IMAGES", "1"))
 guidance_scale = float(os.getenv("GUIDANCE_SCALE", "1.0"))
 
-pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
+pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":"cache"})
 pipe.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
 pipe.compile()
 
@@ -54,7 +54,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 def save_image(img, profile: gr.OAuthProfile | None, metadata: dict):
-    unique_name = str(uuid.uuid4()) + '.png'
+    unique_name = "out/" + str(uuid.uuid4()) + '.png'
     img.save(unique_name)
     gr_user_history.save_image(label=metadata["prompt"], image=img, profile=profile, metadata=metadata)
     return unique_name
@@ -67,6 +67,7 @@ def save_images(image_array, profile: gr.OAuthProfile | None, metadata: dict):
 
 def generate(
     prompt: str,
+    negative_prompt: str,
     seed: int = 0,
     num_inference_steps: int = 4,
     randomize_seed: bool = False,
@@ -84,6 +85,7 @@ def generate(
     start_time = time.time()
     result = pipe(
         prompt=prompt,
+        negative_prompt=negative_prompt,
         width=width,
         height=height,
         guidance_scale=guidance_scale,
@@ -102,7 +104,6 @@ examples = [
     "(masterpiece, best quality, highres), anime art style, pixiv, 1girl, solo, supernova, abstract, abstract background, (bloom, swirling lights, light particles), fire, galaxy, floating, romanticized, blush, looking at viewer, dark red hair, long hair, blue eyes, straight hair, cat ears, medium breasts, mature female, white sweater",
 ]
 
-
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
     gr.DuplicateButton(
@@ -116,7 +117,7 @@ with gr.Blocks(css="style.css") as demo:
                 label="Prompt",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter your prompt",
+                value="(masterpiece, best quality, highres), anime art style, pixiv, 1girl, solo",
                 container=False,
             )
             run_button = gr.Button("Run", scale=0)
@@ -124,6 +125,12 @@ with gr.Blocks(css="style.css") as demo:
             label="Generated images", show_label=False, elem_id="gallery", grid=[2]
         )
     with gr.Accordion("Advanced options", open=False):
+        with gr.Row():
+            negative_prompt = gr.Text(
+	        label="Negative Prompt",
+	        max_lines=1,
+	        value="(worst quality, low quality, lowres), zombie, comic, sketch, blurry, interlocked fingers",
+	    )
         seed = gr.Slider(
             label="Seed",
             minimum=0,
@@ -161,6 +168,7 @@ with gr.Blocks(css="style.css") as demo:
         fn=generate,
         inputs=[
             prompt,
+            negative_prompt,
             seed,
             num_inference_steps,
             randomize_seed
@@ -172,4 +180,4 @@ with gr.Blocks(css="style.css") as demo:
 if __name__ == "__main__":
     demo.queue(api_open=False)
     # demo.queue(max_size=20).launch()
-    demo.launch()
+    demo.launch()