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 |
|