add

inference.py

@@ -0,0 +1,15 @@
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Load the fine-tuned DreamBooth model
+pipe = StableDiffusionPipeline.from_pretrained(
+    "./nyc-ad-model",
+    torch_dtype=torch.float16,
+).to("cuda")  # use "cpu" if no GPU
+
+prompt = "fried chicken advertisement poster: a fried chicken in brooklyn street"
+image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
+
+# Display or save the image
+image.save("output_nyc_ad.png")
+image.show()