nickgardner commited on
Commit
8d7ed76
1 Parent(s): 975d3ac

remove dumb slashes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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