Update app.py

app.py

@@ -332,7 +332,7 @@ def refer_api(prompt):
     
 def model_out_put(init_image,mask_image,prompt,negative_prompt):
     if np.array(mask_image).all():
-        l,m=np.array(init_image).shape
+        l,m=np.array(mask_image).shape
         image =refer_api(prompt).resize((l,m))
         return image
     API_URL = "https://8417-201-238-124-65.ngrok-free.app/api/llm-response"