Update app.py

app.py

@@ -7,6 +7,8 @@ import pandas as pd
 import random
 import string
 import re
+from datetime import datetime
+import pytz
 
 # 2. Constants
 # Translation
@@ -40,7 +42,11 @@ NAMES = pd.DataFrame({
 })
 
 
-device = torch.device('cpu')
+DEVICE = 'gpu' if torch.cuda.is_available() else 'cpu'
+
+device = torch.device(DEVICE)
+
+TZ = pytz.timezone('Europe/Moscow')
 
 # 3. Upload
 # Dictionary
@@ -403,7 +409,7 @@ def translateDisp(text, from_, to, dialect):
   to = "".join(LANGUAGE[LANGUAGE.language == to].token.to_list())
   dialect = "".join(DIALECT[DIALECT.dialect == dialect].short_name.to_list())
 
-  print(f'Input text: {text}')
+  print(f'Input text: {text} - Time: {datetime.now(tz=TZ)}')
     
   text = text.strip()
     
@@ -429,7 +435,7 @@ def translateDisp(text, from_, to, dialect):
   if to == 'krc_Cyrl':
     str_ = fromModel(str_, dialect = dialect)
 
-  print(f'Translated text: {str_}')
+  print(f'Translated text: {str_} - Time: {datetime.now(tz=TZ)}')
     
   return str_