Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ st.title("Movie Genre Predictor")
|
|
12 |
|
13 |
st.subheader("Enter the text you'd like to analyze.")
|
14 |
text = st.text_input('Enter plot of the movie')
|
15 |
-
cast_input = st.text_input("Enter wikipedia ids of the cast in the movie")
|
16 |
|
17 |
model = AutoModelForSequenceClassification.from_pretrained("./checkpoint-36819")
|
18 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
12 |
|
13 |
st.subheader("Enter the text you'd like to analyze.")
|
14 |
text = st.text_input('Enter plot of the movie')
|
15 |
+
cast_input = st.text_input("Enter wikipedia ids of the cast in the movie (Separated by comma)")
|
16 |
|
17 |
model = AutoModelForSequenceClassification.from_pretrained("./checkpoint-36819")
|
18 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|