Spaces:
Runtime error
Runtime error
updated app.py to not train
Browse files
app.py
CHANGED
|
@@ -144,10 +144,10 @@ if os.path.exists(model_state_dict) and os.path.exists(optimizer_state_dict):
|
|
| 144 |
optimizer.load_state_dict(optimizer_state_dict)
|
| 145 |
# Train
|
| 146 |
|
| 147 |
-
for epoch in range(n_epochs):
|
| 148 |
|
| 149 |
-
train(epoch,network,optimizer,train_loader)
|
| 150 |
-
test()
|
| 151 |
|
| 152 |
|
| 153 |
def image_classifier(inp):
|
|
|
|
| 144 |
optimizer.load_state_dict(optimizer_state_dict)
|
| 145 |
# Train
|
| 146 |
|
| 147 |
+
#for epoch in range(n_epochs):
|
| 148 |
|
| 149 |
+
# train(epoch,network,optimizer,train_loader)
|
| 150 |
+
# test()
|
| 151 |
|
| 152 |
|
| 153 |
def image_classifier(inp):
|