Tonic commited on
Commit
d153be5
Β·
verified Β·
1 Parent(s): 0733b82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
  model.to(device)
19
 
20
  # Load spaCy model for dependency parsing
21
- os.system('python -m spacy download la_core_web_lg')
22
- nlp = spacy.load("la_core_web_lg") ## latin-spacy
23
 
24
  # Function for generating text and tokenizing
25
  def historical_generation(prompt, max_new_tokens=600):
 
18
  model.to(device)
19
 
20
  # Load spaCy model for dependency parsing
21
+ os.system('python -m spacy download en_core_web_sm')
22
+ nlp = spacy.load("en_core_web_sm") ## add latin-spacy
23
 
24
  # Function for generating text and tokenizing
25
  def historical_generation(prompt, max_new_tokens=600):