Spaces:
Build error
Build error
Commit
·
ee65817
1
Parent(s):
c5c3656
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def predict(video):
|
|
37 |
layers.Dropout(DROPOUT_RATE),
|
38 |
layers.Dense(64, activation='relu'),
|
39 |
layers.Dropout(DROPOUT_RATE),
|
40 |
-
layers.Dense(2,
|
41 |
])
|
42 |
|
43 |
lossFunction = K.losses.SparseCategoricalCrossentropy(from_logits=False)
|
|
|
37 |
layers.Dropout(DROPOUT_RATE),
|
38 |
layers.Dense(64, activation='relu'),
|
39 |
layers.Dropout(DROPOUT_RATE),
|
40 |
+
layers.Dense(2, activation='softmax')
|
41 |
])
|
42 |
|
43 |
lossFunction = K.losses.SparseCategoricalCrossentropy(from_logits=False)
|