update replicate files

2022-07-02 10:32:17 -04:00
parent e6f1c3ef4c
commit 9cde5b1a60
2 changed files with 4 additions and 2 deletions
--- a/replicate/predict.py
+++ b/replicate/predict.py
@@ -13,10 +13,12 @@ class Predictor(BasePredictor):
            description="Text for generating images.",
        ),
        seed: int = Input(
-            description="Specify a random seed.",
+            description="Specify a random seed."
        ),
        grid_size: int = Input(
            description="Specify the grid size.",
+            ge=1,
+            le=4
        )
    ) -> Path:
        image = self.model.generate_image(text, seed, grid_size=grid_size)