Changed interface code

app.py

@@ -179,7 +179,6 @@ decision_frameworks = {
     "Rule-Based": rule_based_method
 }
 
-# Define the prediction function
 def predict(video_file, video_model_name, audio_model_name, framework_name):
 
     image_processor = VivitImageProcessor.from_pretrained("google/vivit-b-16x2-kinetics400")
@@ -216,17 +215,14 @@ def predict(video_file, video_model_name, audio_model_name, framework_name):
         "Consensus Label": consensus_label
     }
 
-# Create Gradio Interface
 inputs = [
-    gr.inputs.File(label="Upload Video", type="file"),
-    gr.inputs.Dropdown(["video_model_60_acc.pth", "video_model_80_acc.pth"], label="Select Video Model"),
-    gr.inputs.Dropdown(["audio_model_state_dict_6e.pth"], label="Select Audio Model"),
-    gr.inputs.Dropdown(list(decision_frameworks.keys()), label="Select Decision Framework")
+    gr.File(label="Upload Video"),
+    gr.Dropdown(["video_model_60_acc.pth", "video_model_80_acc.pth"], label="Select Video Model"),
+    gr.Dropdown(["audio_model_state_dict_6e.pth"], label="Select Audio Model"),
+    gr.Dropdown(list(decision_frameworks.keys()), label="Select Decision Framework")
 ]
 
-outputs = [
-    gr.outputs.JSON(label="Predictions")
-]
+outputs = gr.JSON(label="Predictions")
 
 iface = gr.Interface(
     fn=predict,