add default arguments to replicate

This commit is contained in:
Brett Kuprel 2022-07-02 10:46:56 -04:00
parent 465b2428de
commit 849e720313

View File

@ -10,19 +10,22 @@ class Predictor(BasePredictor):
def predict( def predict(
self, self,
text: str = Input( text: str = Input(
description="Text for generating images.", description='Text',
default='court sketch of godzilla on trial'
), ),
seed: int = Input( seed: int = Input(
description="Specify a random seed." description='Seed',
default=6
), ),
grid_size: int = Input( grid_size: int = Input(
description="Specify the grid size.", description='Grid Size',
ge=1, ge=1,
le=3 le=3,
default=3
) )
) -> Path: ) -> Path:
image = self.model.generate_image(text, seed, grid_size=grid_size) image = self.model.generate_image(text, seed, grid_size=grid_size)
out_path = Path(tempfile.mkdtemp()) / "output.png" out_path = Path(tempfile.mkdtemp()) / 'output.png'
image.save(str(out_path)) image.save(str(out_path))
return out_path return out_path