updates to empty cache

main.py

@@ -20,6 +20,8 @@ app = FastAPI()
 
 @app.get("/generate")
 def generate_image(prompt, inference_steps, model):
+    torch.cuda.empty_cache()
+    
     print(f"Is CUDA available: {torch.cuda.is_available()}")
 
     #model_id = "CompVis/stable-diffusion-v1-4" #stabilityai/stable-diffusion-2-1