chrisjay commited on
Commit
32e3ac6
1 Parent(s): 37e69f6

updated app.py to not train

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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):