unknown commited on
Commit
0433860
1 Parent(s): be549be

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,8 +20,9 @@ def classify_movie_genre(Overview):
20
  logits = torch.FloatTensor(logits)
21
  probs = torch.sigmoid(logits)[0]
22
  return dict(zip(genres, map(float, probs)))
 
23
 
24
- label = gr.Interface.Label(num_top_classes=5) # Use Label() instead of gr.outputs.Label()
25
  iface = gr.Interface(fn=classify_movie_genre, inputs="text", outputs=label)
26
  iface.launch(inline=False)
27
 
 
20
  logits = torch.FloatTensor(logits)
21
  probs = torch.sigmoid(logits)[0]
22
  return dict(zip(genres, map(float, probs)))
23
+
24
 
25
+ label = gr.outputs.Label(num_top_classes=5)
26
  iface = gr.Interface(fn=classify_movie_genre, inputs="text", outputs=label)
27
  iface.launch(inline=False)
28