Update app.py

app.py

@@ -542,6 +542,8 @@ def upscaler(
     log(f'CALL upscaler')
 
     global working
+    global _HEIGHT_
+    global _WIDTH_
 
     if not working:
 
@@ -632,6 +634,8 @@ def generate_random_string(length):
 def add_song_cover_text(img,top_title=None,bottom_title=None):
 
     global working
+    global _HEIGHT_
+    global _WIDTH_
     
     if not working:
 
@@ -1291,6 +1295,8 @@ def handle_generation(h,w,d):
     log(f'CALL handle_generate')
 
     global working
+    global _HEIGHT_
+    global _WIDTH_
 
     if not working:
 
@@ -1331,7 +1337,7 @@ def handle_generation(h,w,d):
             output_type="pil",
             guidance_scale=img_accu,
             num_images_per_prompt=1,
-            num_inference_steps=image_steps,
+            num_inference_steps=50,
             max_sequence_length=seq,
             generator=torch.Generator(device).manual_seed(random.randint(0, MAX_SEED))
         ).images[0]