Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ def predict(model_name, input_img):
|
|
61 |
experiment_dirname="",
|
62 |
)
|
63 |
|
64 |
-
data_args = DataArgs(batch_size=batch_size, color_mode="L"
|
65 |
trainer_args = TrainerArgs(progress_bar_refresh_rate=0)
|
66 |
decode_args = DecodeArgs(
|
67 |
include_img_ids=True,
|
|
|
61 |
experiment_dirname="",
|
62 |
)
|
63 |
|
64 |
+
data_args = DataArgs(batch_size=batch_size, color_mode="L")
|
65 |
trainer_args = TrainerArgs(progress_bar_refresh_rate=0)
|
66 |
decode_args = DecodeArgs(
|
67 |
include_img_ids=True,
|