Spaces:
Sleeping
Sleeping
Dua Rajper commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -134,7 +134,7 @@ def create_and_train_model(
|
|
| 134 |
Dense(64, activation='relu'),
|
| 135 |
Dense(32, activation='relu'),
|
| 136 |
Dense(num_classes, activation='softmax')
|
| 137 |
-
|
| 138 |
|
| 139 |
if optimizer_str.lower() == 'adam':
|
| 140 |
optimizer = Adam(learning_rate=learning_rate)
|
|
|
|
| 134 |
Dense(64, activation='relu'),
|
| 135 |
Dense(32, activation='relu'),
|
| 136 |
Dense(num_classes, activation='softmax')
|
| 137 |
+
])
|
| 138 |
|
| 139 |
if optimizer_str.lower() == 'adam':
|
| 140 |
optimizer = Adam(learning_rate=learning_rate)
|