fix: handle image input types in leffa_predict function

app.py

@@ -46,8 +46,17 @@ print("Model downloaded, ready to serve!")
 def leffa_predict(src_image_path, ref_image_path, control_type):
     assert control_type in [
         "virtual_tryon", "pose_transfer"], "Invalid control type: {}".format(control_type)
-    src_image = Image.open(src_image_path)
-    ref_image = Image.open(ref_image_path)
+    
+    if isinstance(src_image_path, str):
+        src_image = Image.open(src_image_path)
+    else:
+        src_image = src_image_path
+    
+    if isinstance(ref_image_path, str):
+        ref_image = Image.open(ref_image_path)
+    else:
+        ref_image = ref_image_path
+    
     src_image = resize_and_center(src_image, 768, 1024)
     ref_image = resize_and_center(ref_image, 768, 1024)