Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
22 |
-
nlp = spacy.load("
|
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):
|