johnowhitaker commited on
Commit
5e34623
1 Parent(s): 12a98d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -204,9 +204,10 @@ class ImStack(nn.Module):
204
 
205
 
206
 
207
- def generate(text, n_iter):
208
 
209
  lr=0.25 #@param
 
210
  # init_image=None #@param
211
  weight_decay=1e-5 #@param
212
  out_size=540 #@param
 
204
 
205
 
206
 
207
+ def generate(text, n_steps):
208
 
209
  lr=0.25 #@param
210
+ n_iter = int(n_steps)
211
  # init_image=None #@param
212
  weight_decay=1e-5 #@param
213
  out_size=540 #@param