Fixed weighted average method

app.py

@@ -131,8 +131,7 @@ def averaging_method(video_prediction, audio_prediction):
     consensus_label = max(combined_probabilities, key=combined_probabilities.get)
     return consensus_label
 
-def weighted_average_method(video_prediction, audio_prediction):
-    video_weight = 0.88  
+def weighted_average_method(video_prediction, audio_prediction, video_weight): 
     audio_weight = 0.6  
     combined_probabilities = {}
     for label in set(video_prediction) | set(audio_prediction):
@@ -222,7 +221,13 @@ def predict(video_file, video_model_name, audio_model_name, framework_name):
         highest_audio_emotion = max(audio_prediction, key=audio_prediction.get)
         
         framework_function = decision_frameworks[framework_name]
-        consensus_label = framework_function(video_prediction, audio_prediction)
+
+        if framework_function == weighted_average_method and video_model_name == "60% Accuracy":
+            consensus_label = framework_function(video_prediction, audio_prediction, 0.6)
+        elif framework_function == weighted_average_method and video_model_name == "80% Accuracy":
+            consensus_label = framework_function(video_prediction, audio_prediction, 0.88)
+        else:
+            consensus_label = framework_function(video_prediction, audio_prediction)
     
         delete_files_in_directory(delete_directory_path)