Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
# Load the genre prediction model as a pipeline | |
pipe = pipeline(model="Stanford-TH/GenrePrediction", trust_remote_code=True) | |
def classify_movie_genre(description): | |
# Get predictions using the pipeline | |
predictions = pipe(description) | |
return [predictions] | |
# Define the Gradio interface | |
iface = gr.Interface( | |
fn=classify_movie_genre, | |
inputs="text", | |
outputs="dataframe", | |
live=True, | |
title="Genre Prediction" | |
) | |
# Launch the Gradio interface | |
iface.launch(inline=False) | |