leemeng commited on
Commit
6420f07
·
1 Parent(s): 927ae14

feat: add version info in both log and ui

Browse files
Files changed (1) hide show
  1. app.py +14 -2
app.py CHANGED
@@ -20,6 +20,8 @@ logger.setLevel("INFO")
20
 
21
  gr_interface = None
22
 
 
 
23
  @dataclass
24
  class DefaultArgs:
25
  hf_model_name_or_path: str = None
@@ -133,7 +135,10 @@ def generate(
133
 
134
  # TODO: optimize for final few tokens
135
  gen = streamer.generated_text
136
- log.update(dict(generation=gen, time=str(arrow.now("+09:00"))))
 
 
 
137
  logging.info(log)
138
  yield gen
139
 
@@ -149,7 +154,10 @@ def process_feedback(
149
  no_repeat_ngram_size,
150
  ):
151
  log = dict(locals())
152
- log["time"] = str(arrow.now("+09:00"))
 
 
 
153
  logging.info(log)
154
 
155
  if gr_interface:
@@ -165,6 +173,10 @@ with gr.Blocks() as gr_interface:
165
  # left panel
166
  with gr.Column(scale=1):
167
 
 
 
 
 
168
  # generation params
169
  with gr.Box():
170
  gr.Markdown("パラメータ")
 
20
 
21
  gr_interface = None
22
 
23
+ VERSION = "0.0.0-a.1"
24
+
25
  @dataclass
26
  class DefaultArgs:
27
  hf_model_name_or_path: str = None
 
135
 
136
  # TODO: optimize for final few tokens
137
  gen = streamer.generated_text
138
+ log.update(dict(
139
+ generation=gen,
140
+ version=VERSION,
141
+ time=str(arrow.now("+09:00"))))
142
  logging.info(log)
143
  yield gen
144
 
 
154
  no_repeat_ngram_size,
155
  ):
156
  log = dict(locals())
157
+ log.update(dict(
158
+ time=str(arrow.now("+09:00")),
159
+ version=VERSION,
160
+ ))
161
  logging.info(log)
162
 
163
  if gr_interface:
 
173
  # left panel
174
  with gr.Column(scale=1):
175
 
176
+ # metadata
177
+ with gr.Box():
178
+ gr.Markdown(f"バージョン:{VERSION}")
179
+
180
  # generation params
181
  with gr.Box():
182
  gr.Markdown("パラメータ")