Update app.py

app.py

@@ -120,13 +120,13 @@ def generate_image(prompt,
 
 def predict(lyrics, steps=100, seed=42, guidance_scale=6.0):
 
-    print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
+    # print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
     generator = torch.manual_seed(seed)
     summary_1 = generate_summary(lyrics)
     prompt_text_1 = generate_prompt(summary_1[0]['summary_text'])
-    images = generate_image(prompt= prompt_text_1)
+    images = generate_image(prompt= prompt_text_1, generator= generator, num_inference_steps=steps, guidance_scale=guidance_scale)
     # images = ldm_pipeline([prompt], generator=generator, num_inference_steps=steps, eta=0.3, guidance_scale=guidance_scale)["images"]
-    print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
+    # print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
     return images[0]
 
 random_seed = random.randint(0, 2147483647)