Update src/pipeline.py

src/pipeline.py

@@ -61,9 +61,5 @@ def load_pipeline() -> Pipeline:
 
 @torch.no_grad()
 def infer(request: TextToImageRequest, pipeline: Pipeline, generator: Generator) -> Image:
-    try:
-        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
-    return(image)
+    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