Skip to content

Commit b172b22

Browse files
author
David Motsonashvili
committed
format
1 parent 41427fa commit b172b22

File tree

2 files changed

+16
-17
lines changed

2 files changed

+16
-17
lines changed

firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ImagenTests.kt

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -26,21 +26,20 @@ import org.junit.Test
2626

2727
@OptIn(PublicPreviewAPI::class)
2828
class 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
}

firebase-ai/src/main/kotlin/com/google/firebase/ai/type/ImagenReferenceImage.kt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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
*/

0 commit comments

Comments
 (0)