Update app.py

app.py

@@ -14,6 +14,10 @@ import gradio as gr
 from huggingface_hub.keras_mixin import from_pretrained_keras
 from itertools import cycle, islice
 
+
+model = from_pretrained_keras("tareknaous/unet-visual-clustering")
+
+
 #Function that predicts on only 1 sample 
 def predict_sample(image):
   prediction = model.predict(image[tf.newaxis, ...])
@@ -179,7 +183,6 @@ def visual_clustering(cluster_type, num_clusters, num_samples, random_state, med
 
   return fig1, fig2
   
-model= from_pretrained_keras("tareknaous/unet-visual-clustering")
 
 iface = gr.Interface(