Spaces:
Sleeping
Sleeping
Commit ·
db83dcd
1
Parent(s): c450d50
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,6 +60,9 @@ EOS_IDX = vocab.word2idx["<EOS>"]
|
|
| 60 |
UNK_IDX = vocab.word2idx["<UNK>"]
|
| 61 |
|
| 62 |
|
|
|
|
|
|
|
|
|
|
| 63 |
# ------------- model definitions (same as notebook) -------------
|
| 64 |
class Encoder(nn.Module):
|
| 65 |
def __init__(self):
|
|
|
|
| 60 |
UNK_IDX = vocab.word2idx["<UNK>"]
|
| 61 |
|
| 62 |
|
| 63 |
+
print(f"Vocab size loaded: {len(vocab)} words")
|
| 64 |
+
|
| 65 |
+
|
| 66 |
# ------------- model definitions (same as notebook) -------------
|
| 67 |
class Encoder(nn.Module):
|
| 68 |
def __init__(self):
|