Update app.py

app.py

@@ -79,14 +79,18 @@ def generate_random_string(length):
 
 @spaces.GPU(duration=50)
 def Piper(_do):
-    return pipe(
-        _do,
-        height=512,
-        width=768,
-        num_inference_steps=50,
-        max_sequence_length=512,
-        guidance_scale=2
-    )
+    try:
+        retu = pipe(
+            _do,
+            height=512,
+            width=768,
+            num_inference_steps=50,
+            max_sequence_length=512,
+            guidance_scale=2
+        )
+        return retu
+    except:
+        return ""
 
 def infer(prompt_en):
     name = generate_random_string(12)+".png"
@@ -94,7 +98,11 @@ def infer(prompt_en):
         _do = 'photographed realistic scene'
     else:
         _do = f'photographed { prompt_en } realistic scene'
-    Piper(_do).images[0].save(name)
+    output = Piper(_do)
+    if output == "":
+        return output
+    else:
+        output.images[0].save(name)
     return name
 
 css="""