()
| 24 | } |
| 25 | |
| 26 | export function useTensorFlowLoaded() { |
| 27 | const [isLoaded, setLoaded] = React.useState(false); |
| 28 | |
| 29 | React.useEffect(() => { |
| 30 | let isMounted = true; |
| 31 | tf.ready().then(() => { |
| 32 | if (isMounted) { |
| 33 | setLoaded(true); |
| 34 | } |
| 35 | }); |
| 36 | return () => (isMounted = false); |
| 37 | }, []); |
| 38 | |
| 39 | return isLoaded; |
| 40 | } |