liuhaotian commited on
Commit
f8dc7a7
1 Parent(s): bb6245e
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -16,6 +16,8 @@ from typing import Optional
16
 
17
  import warnings
18
 
 
 
19
  from huggingface_hub import hf_hub_download
20
  hf_hub_download = partial(hf_hub_download, library_name="gligen_demo")
21
 
@@ -59,7 +61,8 @@ class Instance:
59
  self.capacity = capacity
60
 
61
  def _log(self, batch_size, instruction, phrase_list):
62
- print(dict(self.counter), f'samples: {batch_size}', f'prompt: {instruction}', f'phrases: {phrase_list}', sep=', ')
 
63
 
64
  def get_model(self, model_type, batch_size, instruction, phrase_list):
65
  if model_type in self.loaded_model_list:
 
16
 
17
  import warnings
18
 
19
+ from datetime import datetime
20
+
21
  from huggingface_hub import hf_hub_download
22
  hf_hub_download = partial(hf_hub_download, library_name="gligen_demo")
23
 
 
61
  self.capacity = capacity
62
 
63
  def _log(self, batch_size, instruction, phrase_list):
64
+ current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
65
+ print(f'[{current_time}] ' + str(dict(self.counter)), f'samples: {batch_size}', f'prompt: {instruction}', f'phrases: {phrase_list}', sep=', ')
66
 
67
  def get_model(self, model_type, batch_size, instruction, phrase_list):
68
  if model_type in self.loaded_model_list: