Matyáš Boháček commited on
Commit
f734927
1 Parent(s): 039e5e3
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def greet(label, video0, video1):
98
  outputs = model(inputs).expand(1, -1, -1)
99
  results = torch.nn.functional.softmax(outputs, dim=2).detach().numpy()[0, 0]
100
 
101
- results = {GLOSS[i]: float(results[i]) for i in range(100)}
102
 
103
  #temp
104
  results = {GLOSS[i]: float(results[i]) for i in range(15)}
 
98
  outputs = model(inputs).expand(1, -1, -1)
99
  results = torch.nn.functional.softmax(outputs, dim=2).detach().numpy()[0, 0]
100
 
101
+ # results = {GLOSS[i]: float(results[i]) for i in range(100)}
102
 
103
  #temp
104
  results = {GLOSS[i]: float(results[i]) for i in range(15)}