Changed prediction output from JSON to HTML

app.py

@@ -199,7 +199,7 @@ def predict(video_file, video_model_name, audio_model_name, framework_name):
 
     delete_directory_path = "./temp/"
 
-    video_path, audio_path = separate_video_audio(video_file)
+    video_path, audio_path = separate_video_audio(video_file.name)
     
     video_prediction = predict_video(video_path, video_model, image_processor)
 
@@ -212,11 +212,15 @@ def predict(video_file, video_model_name, audio_model_name, framework_name):
     framework_function = decision_frameworks[framework_name]
     consensus_label = framework_function(video_prediction, audio_prediction)
     
-    return {
-        "Video Label": highest_video_emotion,
-        "Audio Label": highest_audio_emotion,
-        "Consensus Label": consensus_label
-    }
+    delete_files_in_directory(delete_directory_path)
+    
+    result = f"""
+    <h2>Predictions</h2>
+    <p><strong>Video Label:</strong> {highest_video_emotion}</p>
+    <p><strong>Audio Label:</strong> {highest_audio_emotion}</p>
+    <p><strong>Consensus Label:</strong> {consensus_label}</p>
+    """
+    return result, video_file.name
 
 inputs = [
     gr.Video(label="Upload Video"),
@@ -226,7 +230,7 @@ inputs = [
 ]
 
 outputs = [
-    gr.JSON(label="Predictions")
+    gr.HTML(label="Predictions")
 ]
 
 iface = gr.Interface(