Update app.py

app.py

@@ -45,7 +45,10 @@ def init():
 
     trainer = pl.Trainer()
 
-    return model.model.detach(), text_tokenizer, cond_tokenizer, args, trainer
+    for param in model.parameters():
+        param.requires_grad = False
+
+    return model, text_tokenizer, cond_tokenizer, args, trainer
 
 img = mpimg.imread('figure.png')
 # Output Images