update

app.py

@@ -128,7 +128,10 @@ def generate(
                     yield r[0]
             except:
                 print("")
-        prior_output = r
+        try:
+            prior_output = r
+        except:
+            print("")
 
     decoder_output = decoder_pipeline(
         image_embeddings=prior_output.image_embeddings,
@@ -297,4 +300,4 @@ with gr.Blocks(css="style.css") as demo_with_history:
 #        user_history.render()
 
 if __name__ == "__main__":
-    demo_with_history.queue(max_size=20).launch()
+    demo_with_history.queue(max_size=20).launch()