update replicate files

main
Brett Kuprel 2 years ago
parent a91a259293
commit e6f1c3ef4c
  1. 0
      cog.yaml
  2. 5
      replicate/predict.py
  3. 2
      setup.py

@ -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))

@ -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',

Loading…
Cancel
Save