10
10
import com .google .android .gms .tasks .OnFailureListener ;
11
11
import com .google .android .gms .tasks .OnSuccessListener ;
12
12
import com .google .firebase .ml .common .FirebaseMLException ;
13
- import com .google .firebase .ml .common .modeldownload .FirebaseCloudModelSource ;
14
- import com .google .firebase .ml .common .modeldownload .FirebaseLocalModelSource ;
13
+ import com .google .firebase .ml .common .modeldownload .FirebaseLocalModel ;
15
14
import com .google .firebase .ml .common .modeldownload .FirebaseModelDownloadConditions ;
16
15
import com .google .firebase .ml .common .modeldownload .FirebaseModelManager ;
16
+ import com .google .firebase .ml .common .modeldownload .FirebaseRemoteModel ;
17
17
import com .google .firebase .ml .custom .FirebaseModelDataType ;
18
18
import com .google .firebase .ml .custom .FirebaseModelInputOutputOptions ;
19
19
import com .google .firebase .ml .custom .FirebaseModelInputs ;
@@ -39,31 +39,31 @@ private void configureHostedModelSource() {
39
39
}
40
40
FirebaseModelDownloadConditions conditions = conditionsBuilder .build ();
41
41
42
- // Build a FirebaseCloudModelSource object by specifying the name you assigned the model
42
+ // Build a remote model source object by specifying the name you assigned the model
43
43
// when you uploaded it in the Firebase console.
44
- FirebaseCloudModelSource cloudSource = new FirebaseCloudModelSource .Builder ("my_cloud_model" )
44
+ FirebaseRemoteModel cloudSource = new FirebaseRemoteModel .Builder ("my_cloud_model" )
45
45
.enableModelUpdates (true )
46
46
.setInitialDownloadConditions (conditions )
47
47
.setUpdatesDownloadConditions (conditions )
48
48
.build ();
49
- FirebaseModelManager .getInstance ().registerCloudModelSource (cloudSource );
49
+ FirebaseModelManager .getInstance ().registerRemoteModel (cloudSource );
50
50
// [END mlkit_cloud_model_source]
51
51
}
52
52
53
53
private void configureLocalModelSource () {
54
54
// [START mlkit_local_model_source]
55
- FirebaseLocalModelSource localSource =
56
- new FirebaseLocalModelSource .Builder ("my_local_model" ) // Assign a name to this model
55
+ FirebaseLocalModel localSource =
56
+ new FirebaseLocalModel .Builder ("my_local_model" ) // Assign a name to this model
57
57
.setAssetFilePath ("my_model.tflite" )
58
58
.build ();
59
- FirebaseModelManager .getInstance ().registerLocalModelSource (localSource );
59
+ FirebaseModelManager .getInstance ().registerLocalModel (localSource );
60
60
// [END mlkit_local_model_source]
61
61
}
62
62
63
63
private FirebaseModelInterpreter createInterpreter () throws FirebaseMLException {
64
64
// [START mlkit_create_interpreter]
65
65
FirebaseModelOptions options = new FirebaseModelOptions .Builder ()
66
- .setCloudModelName ("my_cloud_model" )
66
+ .setRemoteModelName ("my_cloud_model" )
67
67
.setLocalModelName ("my_local_model" )
68
68
.build ();
69
69
FirebaseModelInterpreter firebaseInterpreter =
0 commit comments