Skip to content

Commit de8a2bb

Browse files
committed
support promise when create a new styletransfer method
1 parent c55bc68 commit de8a2bb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/StyleTransfer/index.js

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,8 @@ const styleTransfer = (model, videoOrCallback, cb) => {
140140
callback = videoOrCallback;
141141
}
142142

143-
return new StyleTransfer(model, video, callback);
143+
const instance = new StyleTransfer(model, video, callback);
144+
return callback ? instance : instance.ready;
144145
};
145146

146147
export default styleTransfer;

0 commit comments

Comments
 (0)