Update app.py

app.py

@@ -12,7 +12,7 @@ import numpy as np
 import tensorflow as tf
 import gradio as gr
 from huggingface_hub.keras_mixin import from_pretrained_keras
-
+from itertools import cycle, islice
 
 #Function that predicts on only 1 sample 
 def predict_sample(image):
@@ -46,7 +46,6 @@ def create_input_image(data, visualize=False):
   return input
 
 
-model= from_pretrained_keras("tareknaous/unet-visual-clustering")
 
 def get_instances(prediction, data, max_filter_size=1):
   #Adjust format (clusters to be 255 and rest is 0)
@@ -117,8 +116,7 @@ def get_instances(prediction, data, max_filter_size=1):
   return cluster_ids
   
   
-  import gradio as gr
-from itertools import cycle, islice
+
 
 
 def visual_clustering(cluster_type, num_clusters, num_samples, random_state, median_kernel_size, max_kernel_size):
@@ -180,6 +178,8 @@ def visual_clustering(cluster_type, num_clusters, num_samples, random_state, med
   plt.close()
 
   return fig1, fig2
+  
+model= from_pretrained_keras("tareknaous/unet-visual-clustering")
 
 iface = gr.Interface(