Hugging Face's logo

Spaces:
A-M-S
/
movie-genre
Runtime error

App Files Community
A-M-S commited on
Commit
f08cdc3
•
1 Parent(s): 25309d2

Updated models

Browse files
Files changed (4) hide show
  1. app.py +3 -2
  2. models/cast_mlb +0 -0
  3. models/cast_plot_lr +0 -0
  4. models/meta_model +0 -0
app.py CHANGED
@@ -22,8 +22,9 @@ model.to(device)
22
 
23
  lr_model = pickle.load(open("models/cast_plot_lr","rb"))
24
  cast_mlb = pickle.load(open("models/cast_mlb","rb"))
25
- column_names = pickle.load(open("models/column_names","rb"))
26
- top_actors = list(column_names)[11:]
 
27
 
28
  meta_model = pickle.load(open("models/meta_model","rb"))
29
 
22
 
23
  lr_model = pickle.load(open("models/cast_plot_lr","rb"))
24
  cast_mlb = pickle.load(open("models/cast_mlb","rb"))
25
+ #column_names = pickle.load(open("models/column_names","rb"))
26
+ #top_actors = list(column_names)[11:]
27
+ top_actors = list(cast_mlb.classes_)
28
 
29
  meta_model = pickle.load(open("models/meta_model","rb"))
30
 
models/cast_mlb CHANGED
Binary files a/models/cast_mlb and b/models/cast_mlb differ
models/cast_plot_lr CHANGED
Binary files a/models/cast_plot_lr and b/models/cast_plot_lr differ
models/meta_model CHANGED
Binary files a/models/meta_model and b/models/meta_model differ