Update app.py

app.py

@@ -204,9 +204,10 @@ class ImStack(nn.Module):
  
 
 
-def generate(text, n_iter):
+def generate(text, n_steps):
     
     lr=0.25 #@param
+    n_iter = int(n_steps)
     # init_image=None #@param
     weight_decay=1e-5 #@param
     out_size=540 #@param