Update app.py

app.py

@@ -16,7 +16,7 @@ def classify(review):
     logits = inf_session.run([output_name], {input_name: [input_ids]})[0]
     logits = torch.FloatTensor(logits)
     probs = torch.sigmoid(logits)[0]
-    return dict(zip(genres, map(float, probs))) 
+    return dict(zip(types, map(float, probs))) 
 
 
 label = gr.outputs.Label(num_top_classes=5)