DebasishDhal99 commited on
Commit
cdf308a
·
1 Parent(s): e30a12e

Add logger to a background thread

Browse files
Files changed (1) hide show
  1. 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
- log_to_gsheet(polish_place, german_place, "Polish", "German")
 
 
 
 
 
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