Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view

This file was deleted.

45 changes: 0 additions & 45 deletions genai/image_generation/imggen_upscale_with_img.py

This file was deleted.

47 changes: 0 additions & 47 deletions genai/image_generation/imggen_with_txt.py

This file was deleted.

22 changes: 0 additions & 22 deletions genai/image_generation/test_image_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,10 +36,7 @@
import imggen_raw_reference_with_txt_img
import imggen_scribble_ctrl_type_with_txt_img
import imggen_style_reference_with_txt_img
import imggen_subj_refer_ctrl_refer_with_txt_imgs
import imggen_upscale_with_img
import imggen_virtual_try_on_with_txt_img
import imggen_with_txt

os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
Expand All @@ -63,12 +60,6 @@ def output_gcs_uri() -> str:
blob.delete()


def test_img_generation() -> None:
OUTPUT_FILE = os.path.join(RESOURCES, "dog_newspaper.png")
response = imggen_with_txt.generate_images(OUTPUT_FILE)
assert response


def test_img_edit_inpainting_insert_with_mask() -> None:
OUTPUT_FILE = os.path.join(RESOURCES, "fruit_edit.png")
response = imggen_inpainting_insert_mask_with_txt_img.edit_inpainting_insert_mask(OUTPUT_FILE)
Expand Down Expand Up @@ -117,13 +108,6 @@ def test_img_edit_mask_free() -> None:
assert response


def test_img_customization_subject(output_gcs_uri: str) -> None:
response = imggen_subj_refer_ctrl_refer_with_txt_imgs.subject_customization(
output_gcs_uri=output_gcs_uri
)
assert response


def test_img_customization_style(output_gcs_uri: str) -> None:
response = imggen_style_reference_with_txt_img.style_customization(output_gcs_uri=output_gcs_uri)
assert response
Expand All @@ -148,9 +132,3 @@ def test_img_virtual_try_on() -> None:
OUTPUT_FILE = os.path.join(RESOURCES, "man_in_sweater.png")
response = imggen_virtual_try_on_with_txt_img.virtual_try_on(OUTPUT_FILE)
assert response


def test_img_upscale() -> None:
OUTPUT_FILE = os.path.join(RESOURCES, "dog_newspaper.png")
response = imggen_upscale_with_img.upscale_images(OUTPUT_FILE)
assert response
Loading