Spaces:
Sleeping
Sleeping
typo
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def predict (Sentence, Word, LLM, Norm, Layer):
|
|
35 |
|
36 |
data = (Sentence, Word)
|
37 |
emb = lm.extract_representation(data, layer=Layer)
|
38 |
-
with torch.no_grad:
|
39 |
pred = torch.nn.functional.relu(model(emb))
|
40 |
pred_sq = pred.squeeze(0)
|
41 |
pred_round = torch.round(pred_sq, decimals=2)
|
|
|
35 |
|
36 |
data = (Sentence, Word)
|
37 |
emb = lm.extract_representation(data, layer=Layer)
|
38 |
+
with torch.no_grad():
|
39 |
pred = torch.nn.functional.relu(model(emb))
|
40 |
pred_sq = pred.squeeze(0)
|
41 |
pred_round = torch.round(pred_sq, decimals=2)
|