Spaces:
Build error
Build error
Matyáš Boháček
commited on
Commit
•
ada447d
1
Parent(s):
3a0d867
Remove temp
Browse files
app.py
CHANGED
@@ -98,10 +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 |
-
|
102 |
-
# results = {GLOSS[i]: float(results[i]) for i in range(100)}
|
103 |
-
results = {GLOSS[i]: float(results[i]) for i in range(15)}
|
104 |
-
#ende des temps
|
105 |
|
106 |
return results
|
107 |
|
|
|
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 |
return results
|
104 |
|