Spaces:
Sleeping
Sleeping
Commit
·
cdf308a
1
Parent(s):
e30a12e
Add logger to a background thread
Browse files- backend/german_polish.py +7 -1
backend/german_polish.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
2 |
import torch
|
3 |
from backend.logger.logger import log_to_gsheet
|
|
|
4 |
|
5 |
model_id = "DebasishDhal99/polish-to-german-toponym-model-opus-mt-pl-de"
|
6 |
|
@@ -18,5 +19,10 @@ def polish_to_german_translation(polish_place):
|
|
18 |
outputs = model.generate(**inputs, max_length=50)
|
19 |
|
20 |
german_place = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
21 |
-
|
|
|
|
|
|
|
|
|
|
|
22 |
return german_place
|
|
|
1 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
2 |
import torch
|
3 |
from backend.logger.logger import log_to_gsheet
|
4 |
+
import threading
|
5 |
|
6 |
model_id = "DebasishDhal99/polish-to-german-toponym-model-opus-mt-pl-de"
|
7 |
|
|
|
19 |
outputs = model.generate(**inputs, max_length=50)
|
20 |
|
21 |
german_place = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
22 |
+
threading.Thread(
|
23 |
+
target=log_to_gsheet,
|
24 |
+
args=(polish_place, german_place, "Polish", "German"),
|
25 |
+
daemon=True,
|
26 |
+
).start()
|
27 |
+
# log_to_gsheet(polish_place, german_place, "Polish", "German")
|
28 |
return german_place
|