Update app.py

app.py

@@ -115,7 +115,7 @@ def load_data_and_plot(media_type, media_identifier, csv_path, folder_path):
     # Proceed if we found a valid file
     if media_file:
         prediction_label = extract_prediction_label(media_file, media_type)
-        version_number = extract_version_number(media_file)
+        version_number = extract_version_number(media_file, media_type)
 
         # Get the full path to the media file
         media_path = os.path.join(folder_path, media_file)