.

app.py

@@ -272,6 +272,5 @@ with gr.Blocks(css='style.css') as demo:
     ]
     prompt.submit(fn=pipe.run, inputs=inputs, outputs=result)
     run_button.click(fn=pipe.run, inputs=inputs, outputs=result)
-    if device == 'cuda':
-        torch.cuda.empty_cache()
+
 demo.queue().launch()