Update src/pipeline.py

src/pipeline.py

@@ -60,14 +60,11 @@ def load_pipeline() -> Pipeline:
     
     for _ in range(3):
         pipeline(prompt="insensible, timbale, pothery, electrovital, actinogram, taxis, intracerebellar, centrodesmus", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)
-    empty_cache()
+    # empty_cache()
     return pipeline
 
 sample = 1
 @torch.no_grad()
 def infer(request: TextToImageRequest, pipeline: Pipeline, generator: Generator) -> Image:
-    global sample
-    if not sample:
-        sample=1
-        empty_cache()
+    empty_cache()
     return pipeline(request.prompt,generator=generator, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256, height=request.height, width=request.width, output_type="pil").images[0]