Update src/pipeline.py

src/pipeline.py

@@ -61,10 +61,6 @@ def load_pipeline() -> Pipeline:
 @torch.inference_mode()
 def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
     torch.cuda.reset_peak_memory_stats()
-    try:
-        generator = Generator("cuda").manual_seed(request.seed)
-        image=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]
-    except:
-        image = img.open("./RobertML.png")
-        pass
+    generator = Generator("cuda").manual_seed(request.seed)
+    image=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]
     return(image)