Update README.md

README.md

@@ -193,7 +193,7 @@ except Exception as e:
 
 # Initialize the diffusion pipeline
 try:
-    pipe = DiffusionPipeline.from_pretrained(model_name, use_auth_token=API_TOKEN)
+    pipe = FluxPipeline.from_pretrained(model_name, use_auth_token=API_TOKEN)
     pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 except Exception as e:
     print(f"Error initializing pipeline: {e}")
@@ -212,7 +212,7 @@ except Exception as e:
 
 # Save or display the image
 try:
-    image.save("generated_image.png")
+    image.save("floral-hdr.png")
     image.show()
 except Exception as e:
     print(f"Error saving or displaying image: {e}")