Update app.py

app.py

@@ -41,21 +41,21 @@ def train(prompt, init_image, trn_text, trn_steps):
     return "Training is finished !"
    
 def generate(prompt, init_image, trn_text, trn_steps):
-    #init_image = Image.open(init_image).convert("RGB")
-    #init_image = init_image.resize((256, 256))
+    init_image = Image.open(init_image).convert("RGB")
+    init_image = init_image.resize((256, 256))
 
    
-    #res = pipe.train(
-    #    prompt,
-    #    init_image,
-    #    guidance_scale=7.5,
-    #    num_inference_steps=50,
-    #    generator=generator,
-    #    text_embedding_optimization_steps=trn_text,
-    #    model_fine_tuning_optimization_steps=trn_steps)
+    res = pipe.train(
+        prompt,
+        init_image,
+        guidance_scale=7.5,
+        num_inference_steps=50,
+        generator=generator,
+        text_embedding_optimization_steps=trn_text,
+        model_fine_tuning_optimization_steps=trn_steps)
     
-    #with torch.no_grad():
-    #    torch.cuda.empty_cache()
+    with torch.no_grad():
+        torch.cuda.empty_cache()