Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|