Update app.py

app.py

@@ -8,9 +8,7 @@ def classify_movie_genre(description):
     # Get predictions using the pipeline
     predictions = pipe(description)
     
-    # Format the predictions to match Gradio's output expectations
-    formatted_predictions = {genre: score for genre, score in predictions}
-    return formatted_predictions
+    return predictions
 
 # Define the Gradio interface
 iface = gr.Interface(