Itsme5 commited on
Commit
ab0ee58
·
verified ·
1 Parent(s): a8af6be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  from fastapi import FastAPI
2
- from contextlib import asynccontextmanager
3
  from transformers import PreTrainedTokenizerFast
4
  from tokenizers import ByteLevelBPETokenizer
5
  from datasets import load_dataset
 
6
 
7
  app = FastAPI()
8
 
@@ -10,7 +10,7 @@ app = FastAPI()
10
  async def lifespan(app: FastAPI):
11
  # Εκκίνηση διαδικασίας εκπαίδευσης tokenizer
12
  await train_tokenizer()
13
- yield
14
  # Οποιαδήποτε διαδικασία κατά το κλείσιμο της εφαρμογής
15
 
16
  app = FastAPI(lifespan=lifespan)
 
1
  from fastapi import FastAPI
 
2
  from transformers import PreTrainedTokenizerFast
3
  from tokenizers import ByteLevelBPETokenizer
4
  from datasets import load_dataset
5
+ from contextlib import asynccontextmanager
6
 
7
  app = FastAPI()
8
 
 
10
  async def lifespan(app: FastAPI):
11
  # Εκκίνηση διαδικασίας εκπαίδευσης tokenizer
12
  await train_tokenizer()
13
+ yield # Εδώ κάνουμε την διαδικασία που θα τρέξει μετά την ολοκλήρωση του lifespan
14
  # Οποιαδήποτε διαδικασία κατά το κλείσιμο της εφαρμογής
15
 
16
  app = FastAPI(lifespan=lifespan)