Update app.py

app.py

@@ -44,6 +44,8 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_upscaler i
 )
 from datetime import datetime
 
+working = False
+
 model = T5ForConditionalGeneration.from_pretrained("t5-base")
 tokenizer = T5Tokenizer.from_pretrained("t5-base")
 
@@ -654,7 +656,16 @@ def add_song_cover_text(img,artist,song,height,width):
 
 spaces.GPU(300)
 def upscale_image(img,factor):
-    return upscaler(img,upscale_factor=factor)
+    if working:
+        return
+
+    working = True
+    
+    ret = upscaler(img,upscale_factor=factor)
+
+    working = False
+
+    return ret
 
 def all_pipes(pos,neg,artist,song):
 
@@ -1280,7 +1291,12 @@ def translate(txt,to_lang="en",from_lang="auto"):
 def handle_generation(artist,song,lyrics,height,width):
 
     log(f'CALL handle_generate')
-   
+
+    if working:
+        return
+
+    working = True
+    
     pos_artist = re.sub(r"([ \t]){1,}", " ", artist).upper().strip()
     
     pos_song = re.sub(r"([ \t]){1,}", " ", song).lower().strip()
@@ -1315,6 +1331,8 @@ def handle_generation(artist,song,lyrics,height,width):
     name = f'{generate_random_string(8)}.png'
     labeled_img.save(name)
 
+    working = False
+    
     # return names
     return name