Update app.py
Browse files
app.py
CHANGED
|
@@ -30,6 +30,7 @@ class ANNModel(nn.Module):
|
|
| 30 |
x = self.relu2(x)
|
| 31 |
x = self.fc3(x)
|
| 32 |
x = self.softmax(x)
|
|
|
|
| 33 |
|
| 34 |
# Create an instance of your model
|
| 35 |
ann_model = ANNModel()
|
|
|
|
| 30 |
x = self.relu2(x)
|
| 31 |
x = self.fc3(x)
|
| 32 |
x = self.softmax(x)
|
| 33 |
+
return x
|
| 34 |
|
| 35 |
# Create an instance of your model
|
| 36 |
ann_model = ANNModel()
|