Update app.py

app.py

@@ -13,7 +13,7 @@ pipe = OVStableDiffusionPipeline.from_pretrained(
     model_id,
     compile=False,
     ov_config={"CACHE_DIR": ""},
-    torch_dtype=torch.float16,  # More standard dtype for speed
+    torch_dtype=torch.bfloat16,  # More standard dtype for speed
     safety_checker=None,
     use_safetensors=False,
 )
@@ -24,10 +24,8 @@ pipe.reshape(batch_size=batch_size, height=HIGH, width=WIDTH, num_images_per_pro
 pipe.compile()
 
 prompt = ""
-negative_prompt = "EasyNegative, "
-num_inference_steps = 4
 
-def infer(prompt, negative_prompt, num_inference_steps):
+def infer(prompt, negative_prompt="EasyNegative, ", num_inference_steps=4):
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,