PierreHanna commited on
Commit
e38c032
β€’
1 Parent(s): c7e307d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -8,10 +8,13 @@ import sys
8
  import numpy as np
9
  import faiss
10
  import csv
 
 
11
 
12
  # NO GPU
13
  os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
14
- os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'
 
15
 
16
 
17
  def make_bert_preprocess_model(sentence_features, tfhub_handle_preprocess, seq_length=128):
@@ -53,6 +56,8 @@ def process(prompt, lang):
53
  bert_preprocess_model = make_bert_preprocess_model(['my_input'], tfhub_handle_preprocess, seq_length = MAX_LENGTH)
54
  bert_model = hub.KerasLayer(tfhub_handle_encoder)
55
 
 
 
56
  print("Text input : ", prompt)
57
  prompt=[prompt]
58
  text_preprocessed = bert_preprocess_model([np.array(prompt)])
 
8
  import numpy as np
9
  import faiss
10
  import csv
11
+ import datetime
12
+
13
 
14
  # NO GPU
15
  os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
16
+ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
17
+
18
 
19
 
20
  def make_bert_preprocess_model(sentence_features, tfhub_handle_preprocess, seq_length=128):
 
56
  bert_preprocess_model = make_bert_preprocess_model(['my_input'], tfhub_handle_preprocess, seq_length = MAX_LENGTH)
57
  bert_model = hub.KerasLayer(tfhub_handle_encoder)
58
 
59
+ now = datetime.datetime.now()
60
+ print("Current date and time: ", str(now))
61
  print("Text input : ", prompt)
62
  prompt=[prompt]
63
  text_preprocessed = bert_preprocess_model([np.array(prompt)])