Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def create_classifier_model():
|
|
| 26 |
nn.Linear(num_ftrs, 512),
|
| 27 |
nn.ReLU(),
|
| 28 |
nn.Linear(512, 5),
|
| 29 |
-
nn.LogSoftmax(dim=1)
|
| 30 |
return model
|
| 31 |
|
| 32 |
@st.cache_resource
|
|
|
|
| 26 |
nn.Linear(num_ftrs, 512),
|
| 27 |
nn.ReLU(),
|
| 28 |
nn.Linear(512, 5),
|
| 29 |
+
nn.LogSoftmax(dim=1))
|
| 30 |
return model
|
| 31 |
|
| 32 |
@st.cache_resource
|