Cleaned output labels

app.py

@@ -199,25 +199,24 @@ def predict(video_file, video_model_name, audio_model_name, framework_name):
 
     delete_directory_path = "./temp/"
 
-    # Separate video and audio
     video_path, audio_path = separate_video_audio(video_file.name)
     
-    # Predict video
     video_prediction = predict_video(video_path, video_model, image_processor)
+
+    highest_video_emotion = max(video_prediction, key=video_prediction.get)
     
-    # Predict audio
     audio_prediction = preprocess_and_predict_audio(audio_path, audio_model, audio_processor)
+
+    highest_audio_emotion = max(audio_prediction, key=audio_prediction.get)
     
-    # Use selected decision framework
     framework_function = decision_frameworks[framework_name]
     consensus_label = framework_function(video_prediction, audio_prediction)
     
-    # Clean up the temporary files
     delete_files_in_directory(delete_directory_path)
     
     return {
-        "Video Predictions": video_prediction,
-        "Audio Predictions": audio_prediction,
+        "Video Label": highest_video_emotion,
+        "Audio Label": highest_audio_emotion,
         "Consensus Label": consensus_label
     }