Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import pipeline
+import pandas as pd 
 
 # Load the genre prediction model as a pipeline
 pipe = pipeline(model="Stanford-TH/GenrePrediction", trust_remote_code=True)
@@ -8,7 +9,7 @@ def classify_movie_genre(description):
     # Get predictions using the pipeline
     predictions = pipe(description)
     
-    return [predictions]
+    return pd.DataFrame([predictions])
 
 # Define the Gradio interface
 iface = gr.Interface(