takarajordan
commited on
Update train.py
Browse files
train.py
CHANGED
@@ -320,7 +320,7 @@ def train_model(image_path, message, epochs=600):
|
|
320 |
if __name__ == "__main__":
|
321 |
input_image = "steno_2(1).jpg"
|
322 |
output_image = "decode_me_3.png"
|
323 |
-
secret_message = "
|
324 |
|
325 |
# Train model
|
326 |
model = train_model(input_image, secret_message)
|
|
|
320 |
if __name__ == "__main__":
|
321 |
input_image = "steno_2(1).jpg"
|
322 |
output_image = "decode_me_3.png"
|
323 |
+
secret_message = ""
|
324 |
|
325 |
# Train model
|
326 |
model = train_model(input_image, secret_message)
|