Update app.py

app.py

@@ -37,11 +37,15 @@ def extract_prediction_label(filename, media_type):
     if media_type == "video":
         return filename.split('_')[3]  # Extracting the label from video filenames
     elif media_type == "image":
-        return filename.split('_')[4]  # Extracting the label from image filenames
+        return filename.split('_')[3]  # Extracting the label from image filenames
 
 # Function to extract the version number from filenames
-def extract_version_number(filename):
-    version_with_v = filename.split('_')[1]  # Extracting something like 'v1'
+def extract_version_number(filename, media_type):
+    if media_type == "video":
+        version_with_v = filename.split('_')[1]  # For videos, version is at index 1
+    elif media_type == "image":
+        version_with_v = filename.split('_')[2]  # For images, version is at index 2
+    
     version_number = version_with_v[1:]  # Remove the 'v' and keep only the number
     return version_number