Update README.md

README.md

@@ -64,22 +64,22 @@ def predict(image, mask_image, strength=1, steps=100):
       negative_prompt=negative_prompt,#Negative prompt
       strength=strength, #Strength (0.1 - 1)
       num_inference_steps=steps, #Generation steps
-      image=image, #PIL obj
-      mask_image=mask_image, #PIL obj
+      image=image, #PIL Image obj
+      mask_image=mask_image, #PIL Image obj
       generator=generator, #Generator
       num_images_per_prompt=1, #Image per predict
   ).images
 
   return result
 
-#Open image preprocessing and other operations
+#Open image, pre-processing and other operations, must get image, mask : PIL Image Obj
 image = ... #PIL Image 512x512
 mask = ... #PIL black and white mask image 512x512
 
 #Forward and getting results
 image = predict(image,mask)[0] #result <- [array num_images_per_prompt](take the first image index: 0)
 
-image #PIL Obj with the result of model generation
+image #PIL Image Obj with the result of model generation
 
 ```