Rimi98 commited on
Commit
3ce5c45
·
1 Parent(s): b893b69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ output_name = inf_session.get_outputs()[0].name
12
 
13
  classes = ['Art', 'Astrology', 'Biology', 'Chemistry', 'Economics', 'History', 'Literature', 'Philosophy', 'Physics', 'Politics', 'Psychology', 'Sociology']
14
 
15
- def classify(review):
16
  input_ids = token(review)['input_ids'][:512]
17
  logits = inf_session.run([output_name],{input_name : [input_ids]})[0]
18
  logits = torch.FloatTensor(logits)
@@ -21,7 +21,7 @@ def classify(review):
21
  return dict(zip(classes,map(float,probs)))
22
 
23
  label = gr.outputs.Label(num_top_classes=5)
24
- iface = gr.Interface(fn=classify,inputs='text',outputs = label)
25
  iface.launch(inline=False)
26
 
27
 
 
12
 
13
  classes = ['Art', 'Astrology', 'Biology', 'Chemistry', 'Economics', 'History', 'Literature', 'Philosophy', 'Physics', 'Politics', 'Psychology', 'Sociology']
14
 
15
+ def classify(vid,review):
16
  input_ids = token(review)['input_ids'][:512]
17
  logits = inf_session.run([output_name],{input_name : [input_ids]})[0]
18
  logits = torch.FloatTensor(logits)
 
21
  return dict(zip(classes,map(float,probs)))
22
 
23
  label = gr.outputs.Label(num_top_classes=5)
24
+ iface = gr.Interface(fn=classify,inputs=[gr.Video(),'text'],outputs = label)
25
  iface.launch(inline=False)
26
 
27