Update app.py

app.py

@@ -5,13 +5,7 @@ import torch, json
 
 token = AutoTokenizer.from_pretrained('distilroberta-base')
 
-types = ['toxic',
-  'severe_toxic',
-  'obscene',
-  'threat',
-  'insult',
-  'identity_hate',
-  'positive']
+types = ['Toxic','Severe_toxic','Obscene','Threat','Insult','Identity_hate','Positive']
 
 inf_session = onnxruntime.InferenceSession('classifier-quantized.onnx')
 input_name = inf_session.get_inputs()[0].name
@@ -25,6 +19,6 @@ def classify(review):
     return dict(zip(genres, map(float, probs))) 
 
 
-label = gr.outputs.Label(num_top_classes=3)
-iface = gr.Interface(fn=classify,inputs='text',outputs = label)
+label = gr.outputs.Label(num_top_classes=5)
+iface = gr.Interface(fn=classify_movie_genre, inputs="text", outputs=label)
 iface.launch(inline=False)