A-M-S commited on
Commit
aa72060
1 Parent(s): 41e5d76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'