Johannes Kolbe commited on
Commit
8716e2b
1 Parent(s): b65403d

adapt keras-io model location

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ classes = [
28
  "truck",
29
  ]
30
 
31
- clustering_model = from_pretrained_keras("johko/semantic-image-clustering")
32
 
33
  # Get the cluster probability distribution of the input images.
34
  clustering_probs = clustering_model.predict(x_data, batch_size=500, verbose=1)
 
28
  "truck",
29
  ]
30
 
31
+ clustering_model = from_pretrained_keras("keras-io/semantic-image-clustering")
32
 
33
  # Get the cluster probability distribution of the input images.
34
  clustering_probs = clustering_model.predict(x_data, batch_size=500, verbose=1)