Update app.py

app.py

@@ -25,6 +25,7 @@ def generate(prompt, width=832, height=832, num_inference_steps=30, lora_id=None
         pipe.unload_lora_weights()
         pipe.load_lora_weights(lora_id.strip())
     pipe.to("cuda")
+    torch.cuda.empty_cache()
     try:
         output = pipe(
             prompt=prompt,