update replicate files

This commit is contained in:
Brett Kuprel 2022-07-02 10:05:16 -04:00
parent a91a259293
commit e6f1c3ef4c
3 changed files with 5 additions and 2 deletions

View File

View File

@ -15,8 +15,11 @@ class Predictor(BasePredictor):
seed: int = Input(
description="Specify a random seed.",
),
grid_size: int = Input(
description="Specify the grid size.",
)
) -> Path:
image = self.model.generate_image(text, seed, grid_size=3)
image = self.model.generate_image(text, seed, grid_size=grid_size)
out_path = Path(tempfile.mkdtemp()) / "output.png"
image.save(str(out_path))

View File

@ -5,7 +5,7 @@ setuptools.setup(
name='min-dalle',
description = 'min(DALL·E)',
long_description=(Path(__file__).parent / "README").read_text(),
version='0.2.9',
version='0.2.10',
author='Brett Kuprel',
author_email='brkuprel@gmail.com',
url='https://github.com/kuprel/min-dalle',