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