Update app.py

app.py

@@ -103,7 +103,7 @@ def load_model():
     pipeline_ = AutoPipelineForInpainting.from_pretrained("kandinsky-community/kandinsky-2-2-decoder-inpaint", torch_dtype=torch.float16).to("cuda")
     return pipeline_
 
-# @st.cache_resource
+@st.cache_resource
 def prompt_improvment(pre_prompt):
 
     enhancement="Please use details from the prompt mentioned above, focusing only what user is thinking with the prompt and also add 8k resolution. Its a request only provide image description and brief prompt no other text."
@@ -165,16 +165,16 @@ def model_out_put(init_image,mask_image,prompt,negative_prompt):
     # output_image.show()
     return output_image
 
-# def model_out_put(init_image, mask_image, prompt, negative_prompt):
-#     # Run the inpainting pipeline
-#     pipeline_ = load_model()
-#     image = pipeline_(
-#         prompt=prompt, 
-#         negative_prompt=negative_prompt, 
-#         image=init_image, 
-#         mask_image=mask_image
-#     ).images[0]
-#     return image
+def model_out_put(init_image, mask_image, prompt, negative_prompt):
+    # Run the inpainting pipeline
+    pipeline_ = load_model()
+    image = pipeline_(
+        prompt=prompt, 
+        negative_prompt=negative_prompt, 
+        image=init_image, 
+        mask_image=mask_image
+    ).images[0]
+    return image
 
 @st.cache_resource
 def multimodel():