sudaisdev commited on
Commit
b5d8bea
·
verified ·
1 Parent(s): 94574ed

Create model.py

Browse files
Files changed (1) hide show
  1. model.py +19 -0
model.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # model.py
2
+ # CharRNN model class — Hugging Face par upload karne ke liye
3
+ # Yeh file user import karega taake model load ho sake
4
+
5
+ import torch
6
+ import torch.nn as nn
7
+
8
+ class CharRNN(nn.Module):
9
+ def __init__(self, vocab_size, embed_size=64, hidden_size=256):
10
+ super().__init__()
11
+ self.embedding = nn.Embedding(vocab_size, embed_size)
12
+ self.rnn = nn.RNN(embed_size, hidden_size, batch_first=True)
13
+ self.fc = nn.Linear(hidden_size, vocab_size)
14
+
15
+ def forward(self, x):
16
+ emb = self.embedding(x)
17
+ out, _ = self.rnn(emb)
18
+ logits = self.fc(out)
19
+ return logits