diff --git a/Stable_diff_text2image.py b/Stable_diff_text2image.py
index 237cd0f..9a081c4 100644
--- a/Stable_diff_text2image.py
+++ b/Stable_diff_text2image.py
@@ -6,7 +6,7 @@ from towhee.operator import PyOperator
 log = logging.getLogger(PyOperator)
 
 class StableDiffusion(PyOperator):
-    def __init__(self,model_id='stabilityai/stable-diffusion-2-1',pipe='StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)'):
+    def __init__(self,model_id='stabilityai/stable-diffusion-2-1',pipe='StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32'):
         self._model_id=model_id
         self.pipe = pipe