Added classes count

app.py

@@ -55,6 +55,16 @@ vision_model = TFCLIPVisionModel.from_pretrained(
     cache_dir=PRE_TRAINED_MODELS_FOLDER)
 vision_processor = CLIPImageProcessor.from_pretrained(clip_model_id)
 
+genre_classes_path = os.path.join(ROOT_FOLDER,'genre_class.txt')
+# TSV headers [id, class]
+genre_classes_df = pd.read_csv(genre_classes_path, sep = ' ',  header=None)
+# print(genre_train_df.iloc[:,1])
+genre_classes = []
+for index, row in genre_classes_df.iterrows():
+    genre_classes.append(row[1])
+# print(genre_classes)
+classes_count = len(genre_classes)
+
 base_learning_rate = 0.0001
 steps_per_execution = 200
 
@@ -162,15 +172,6 @@ model.load_weights(latest_weights)
 # result = model.predict(np_input)
 # print(result.flatten())
 
-genre_classes_path = os.path.join(ROOT_FOLDER,'genre_class.txt')
-# TSV headers [id, class]
-genre_classes_df = pd.read_csv(genre_classes_path, sep = ' ',  header=None)
-# print(genre_train_df.iloc[:,1])
-genre_classes = []
-for index, row in genre_classes_df.iterrows():
-    genre_classes.append(row[1])
-# print(genre_classes)
-
 import gradio as gr
 
 def process_image(input_image):