wip

Signed-off-by: Raphael <[email protected]>

app.py

@@ -114,11 +114,17 @@ def merge_selfie_and_background(selfie, background, mask):
 
 
 def demo(threshold, image, prompt, num_inference_steps):
-    image = image_preprocess(image)
-    mask = extract_selfie_mask(threshold, image)
-    background = generate_background(prompt, num_inference_steps,
-                                     image.shape[0], image.shape[1])
-    output = merge_selfie_and_background(image, background, mask)
+    LOG.info("Processing image")
+    try:
+        image = image_preprocess(image)
+        mask = extract_selfie_mask(threshold, image)
+        background = generate_background(prompt, num_inference_steps,
+                                         image.shape[0], image.shape[1])
+        output = merge_selfie_and_background(image, background, mask)
+    except Exception as e:
+        LOG.error("Some unexpected error occured")
+        LOG.exception(e)
+        raise
     return output
 
 

requirements.txt

@@ -10,3 +10,4 @@ ftfy
 accelerate
 torch
 numpy
+hf_transfer