Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer
 import torch
 import json
 
-tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
+tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 
 with open("genre_types_encoded.json", "r") as fp:
     encode_genre_types = json.load(fp)
@@ -33,6 +33,6 @@ def classify_movie_genre(summary):
     
     return dict(zip(genres, map(float, probs))) 
 
-label = gr.Label(num_top_classes=10)
+label = gr.Label(num_top_classes=5)
 iface = gr.Interface(fn=classify_movie_genre, inputs="text", outputs=label)
 iface.launch(inline=False)