dreibh commited on
Commit
b62eb97
·
verified ·
1 Parent(s): 76f65c9

Added ECG information (25 mm/sec, 1 mV/10 mm) to plot.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -154,6 +154,7 @@ def predict(numberOfECGs: int = 1,
154
  # ====== Create a list of image/label tuples for gradio.Gallery ==========
155
  plotList = []
156
  ecgNumber = 1
 
157
  for result in Sessions[request.session_hash].Results:
158
 
159
  # ====== Plot ECG =====================================================
@@ -166,7 +167,7 @@ def predict(numberOfECGs: int = 1,
166
  # ------ ECG-12 -------------------------------------------------------
167
  if ecgType == deepfakeecg.DATA_ECG12:
168
  ecg_plot.plot(result,
169
- title = 'ECG-12',
170
  sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
171
  lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6', 'III', 'aVR', 'aVL', 'aVF' ],
172
  lead_order = [0, 1, 8, 9, 10, 11, 2, 3, 4, 5, 6, 7],
@@ -174,7 +175,7 @@ def predict(numberOfECGs: int = 1,
174
  # ------ ECG-8 --------------------------------------------------------
175
  else:
176
  ecg_plot.plot(result,
177
- title = 'ECG-8',
178
  sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
179
  lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6' ],
180
  lead_order = [0, 1, 2, 3, 4, 5, 6, 7],
 
154
  # ====== Create a list of image/label tuples for gradio.Gallery ==========
155
  plotList = []
156
  ecgNumber = 1
157
+ info = '25 mm/sec, 1 mV/10 mm'
158
  for result in Sessions[request.session_hash].Results:
159
 
160
  # ====== Plot ECG =====================================================
 
167
  # ------ ECG-12 -------------------------------------------------------
168
  if ecgType == deepfakeecg.DATA_ECG12:
169
  ecg_plot.plot(result,
170
+ title = 'ECG-12' + info,
171
  sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
172
  lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6', 'III', 'aVR', 'aVL', 'aVF' ],
173
  lead_order = [0, 1, 8, 9, 10, 11, 2, 3, 4, 5, 6, 7],
 
175
  # ------ ECG-8 --------------------------------------------------------
176
  else:
177
  ecg_plot.plot(result,
178
+ title = 'ECG-8' + info,
179
  sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
180
  lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6' ],
181
  lead_order = [0, 1, 2, 3, 4, 5, 6, 7],