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