update replicate file

main
Brett Kuprel 2 years ago
parent e087c53482
commit 72cd1501f2
  1. 4
      replicate/predict.py

@ -13,10 +13,10 @@ class Predictor(BasePredictor):
description="Text for generating images.",
),
seed: int = Input(
description="Specify the seed.",
description="Specify a random seed.",
),
) -> Path:
image = self.model.generate_image(text, seed)
image = self.model.generate_image(text, seed, grid_size=3)
out_path = Path(tempfile.mkdtemp()) / "output.png"
image.save(str(out_path))

Loading…
Cancel
Save