BlinkDL commited on
Commit
4c64617
1 Parent(s): a123817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -160,6 +160,8 @@ def evaluate_eng(
160
  out_str += tmp
161
  yield out_str.strip()
162
  out_last = i + 1
 
 
163
 
164
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
165
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -210,6 +212,8 @@ def evaluate_chn(
210
  out_str += tmp
211
  yield out_str.strip()
212
  out_last = i + 1
 
 
213
 
214
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
215
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -260,6 +264,8 @@ def evaluate_wyw(
260
  out_str += tmp
261
  yield out_str.strip()
262
  out_last = i + 1
 
 
263
 
264
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
265
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
160
  out_str += tmp
161
  yield out_str.strip()
162
  out_last = i + 1
163
+ if '\n\n' in out_str:
164
+ break
165
 
166
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
167
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
212
  out_str += tmp
213
  yield out_str.strip()
214
  out_last = i + 1
215
+ if '\n\n' in out_str:
216
+ break
217
 
218
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
219
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
264
  out_str += tmp
265
  yield out_str.strip()
266
  out_last = i + 1
267
+ if '\n\n' in out_str:
268
+ break
269
 
270
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
271
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")