Spaces:
Runtime error
Runtime error
Commit
·
8d7ed76
1
Parent(s):
975d3ac
remove dumb slashes
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ subprocess.run(["pip", "install", model_url])
|
|
14 |
MAX_LEN = 350
|
15 |
|
16 |
tokenizer = get_tokenizer('spacy', language='en_core_web_sm')
|
17 |
-
vocab = torch.load(hf_hub_download(repo_id="nickgardner/chatbot
|
18 |
filename="vocab.pth"))
|
19 |
vocab_token_dict = vocab.get_stoi()
|
20 |
pad_token = vocab_token_dict['<pad>']
|
@@ -29,7 +29,7 @@ N = 6
|
|
29 |
src_vocab = len(vocab)
|
30 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
31 |
model = Transformer(len(vocab), len(vocab), d_model, N, heads).to(device)
|
32 |
-
model.load_state_dict(hf_hub_download(repo_id="nickgardner/chatbot
|
33 |
filename="alpaca_train_380_epoch.pt"))
|
34 |
model.eval()
|
35 |
|
|
|
14 |
MAX_LEN = 350
|
15 |
|
16 |
tokenizer = get_tokenizer('spacy', language='en_core_web_sm')
|
17 |
+
vocab = torch.load(hf_hub_download(repo_id="nickgardner/chatbot",
|
18 |
filename="vocab.pth"))
|
19 |
vocab_token_dict = vocab.get_stoi()
|
20 |
pad_token = vocab_token_dict['<pad>']
|
|
|
29 |
src_vocab = len(vocab)
|
30 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
31 |
model = Transformer(len(vocab), len(vocab), d_model, N, heads).to(device)
|
32 |
+
model.load_state_dict(hf_hub_download(repo_id="nickgardner/chatbot",
|
33 |
filename="alpaca_train_380_epoch.pt"))
|
34 |
model.eval()
|
35 |
|