Update app.py

app.py

@@ -87,9 +87,24 @@ def model_single_out(prompt):
     return image
 
 def model_out_put(init_image,mask_image,prompt,negative_prompt):
-    pipeline_ = load_model()
-    image = pipeline_(prompt=prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask_image).images[0]
-    return image
+    API_URL = "https://d5e8-2405-201-802c-f859-c1ff-35e2-34fb-b340.ngrok-free.app/api/llm-response"
+    initial_image_base64 = numpy_to_list(np.array(init_image))
+    mask_image_base64 = numpy_to_list(np.array(mask_image))
+    payload = {
+        "prompt": prompt,  # Replace with your desired prompt
+        "initial_img": initial_image_base64,
+        "masked_img": mask_image_base64,
+        "negative_prompt": negative_prompt # Replace with your negative prompt
+    }
+    response = requests.post(API_URL, json=payload)
+    response_data = response.json()
+    output_image_base64 = response_data.get("img", "")
+
+    output_image=np.array(output_image_base64,dtype=np.uint8)
+
+    output_image = Image.fromarray(output_image)
+    # output_image.show()
+    return output_image
 
 @st.cache_resource
 def multimodel():