Update app.py

app.py

@@ -39,7 +39,7 @@ preprocess = Compose([
 ])
 
 @st.cache_data
-def clip_optimized_latent(text, seed, iterations=25, lr=1e-2):
+def clip_optimized_latent(text, seed, iterations=25, lr=0.1):
     seed = int(seed)
     text_input = clip.tokenize([text]).to(device)
 
@@ -48,7 +48,7 @@ def clip_optimized_latent(text, seed, iterations=25, lr=1e-2):
     latent_vector.requires_grad = True
     latent_vector = [latent_vector]*model.get_max_latents()
     params = [torch.nn.Parameter(latent_vector[i], requires_grad=True) for i in range(len(latent_vector))]
-    optimizer = Adam(params, lr=lr, betas=(0.9, 0.999))
+    optimizer = Adam(params, lr=lr)
 
     #with torch.no_grad():
     #    text_features = clip_model.encode_text(text_input)