File tree Expand file tree Collapse file tree 2 files changed +16
-17
lines changed
androidTest/kotlin/com/google/firebase/ai
main/kotlin/com/google/firebase/ai/type Expand file tree Collapse file tree 2 files changed +16
-17
lines changed Original file line number Diff line number Diff line change @@ -26,21 +26,20 @@ import org.junit.Test
2626
2727@OptIn(PublicPreviewAPI ::class )
2828class ImagenTests {
29- @Test
30- fun testGenerateAndEditImage () {
31- val imageGenerationModel = FirebaseAI .getInstance(app()).imagenModel(" imagen-3.0-generate-002" )
32- val imageEditingModel = FirebaseAI .getInstance(app()).imagenModel(" imagen-3.0-capability-001" )
29+ @Test
30+ fun testGenerateAndEditImage () {
31+ val imageGenerationModel = FirebaseAI .getInstance(app()).imagenModel(" imagen-3.0-generate-002" )
32+ val imageEditingModel = FirebaseAI .getInstance(app()).imagenModel(" imagen-3.0-capability-001" )
3333
34- runBlocking {
35- val catImage = imageGenerationModel.generateImages(" A cat" ).images.first()
36- val editedCatImage = imageEditingModel.editImage(
37- listOf (ImagenRawImage (catImage), ImagenBackgroundMask ()),
38- " A cat flying through space" ,
39- ImagenEditingConfig (
40- ImagenEditMode .INPAINT_INSERTION
41- )
42- )
43- assert (editedCatImage.images.size == 1 )
44- }
34+ runBlocking {
35+ val catImage = imageGenerationModel.generateImages(" A cat" ).images.first()
36+ val editedCatImage =
37+ imageEditingModel.editImage(
38+ listOf (ImagenRawImage (catImage), ImagenBackgroundMask ()),
39+ " A cat flying through space" ,
40+ ImagenEditingConfig (ImagenEditMode .INPAINT_INSERTION )
41+ )
42+ assert (editedCatImage.images.size == 1 )
4543 }
44+ }
4645}
Original file line number Diff line number Diff line change @@ -80,8 +80,8 @@ internal constructor(
8080 * @param referenceId the reference ID for this image, to be referenced in the prompt
8181 * @param [enableComputation] requests that the reference image be generated serverside instead of
8282 * provided
83- * @param superpixelRegionSize if type is COLOR_SUPERPIXEL and [enableComputation] is true, this will
84- * control the size of each superpixel region in pixels for the generated referenced image
83+ * @param superpixelRegionSize if type is COLOR_SUPERPIXEL and [enableComputation] is true, this
84+ * will control the size of each superpixel region in pixels for the generated referenced image
8585 * @param superpixelRuler if type is COLOR_SUPERPIXEL and [enableComputation] is true, this will
8686 * control the superpixel smoothness factor for the generated referenced image
8787 */
You can’t perform that action at this time.
0 commit comments