dennistrujillo commited on
Commit
9f5ee14
·
1 Parent(s): f0d85bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -10,11 +10,12 @@ import numpy as np
10
  import gradio as gr
11
 
12
  def generate_frames(ffile,dfile):
13
- frd = FrameReaderDataset(ffile=ffile,dfile=dfile)
14
- frames=frd.get_frames()
15
- np.save("frames.npy",frames)
16
- return "frames.npy"
17
-
 
18
  interface=gr.Interface(fn=generate_frames, inputs=[gr.File(label="frame file"),gr.File(label="dark file")],
19
  outputs=gr.outputs.File("peaks.npy"))
20
  interface.launch()
 
10
  import gradio as gr
11
 
12
  def generate_frames(ffile,dfile):
13
+ dataset = FrameReaderDataset(ffile=ffile,dfile=dfile)
14
+ patches = dataset.get_peaks_skimage()
15
+ normalized_patches = dataset.normalize_patches(patches)
16
+ np.save("normalized_patches.npy",normalized_patches)
17
+ return normalized_patches
18
+
19
  interface=gr.Interface(fn=generate_frames, inputs=[gr.File(label="frame file"),gr.File(label="dark file")],
20
  outputs=gr.outputs.File("peaks.npy"))
21
  interface.launch()