kenmirrors/ente
1
0
Fork 0
mirror of https://github.com/ente-io/ente.git synced 2025-08-08 07:28:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

[mob] Fix state refresh when model is being downloaded

Browse Source
This commit is contained in:
Neeraj Gupta 2024-08-16 15:13:01 +05:30
parent 78fc2aec88
commit 1731aeb87d
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mobile/lib/ui/settings/machine_learning_settings_page.dart
View File

@ -35,12 +35,20 @@ class MachineLearningSettingsPage extends StatefulWidget {
class _MachineLearningSettingsPageState
extends State<MachineLearningSettingsPage> {
final EnteWakeLock _wakeLock = EnteWakeLock();
Timer? _timer;
@override
void initState() {
super.initState();
_wakeLock.enable();
MachineLearningController.instance.forceOverrideML(turnOn: true);
if (!MLService.instance.areModelsDownloaded) {
_timer = Timer.periodic(const Duration(seconds: 10), (timer) {
if (mounted) {
setState(() {});
}
});
}
}
@override
@ -48,6 +56,7 @@ class _MachineLearningSettingsPageState
super.dispose();
_wakeLock.disable();
MachineLearningController.instance.forceOverrideML(turnOn: false);
_timer?.cancel();
}
@override