Update app.py

app.py

@@ -5,12 +5,12 @@ os.system("pip -qq install torchvision")
 os.system("pip -qq install pandas")
 os.system("pip -qq install scikit-image")
 
-from dataset import FrameReaderDataset
+from dataset import FrameDataset
 import numpy as np
 import gradio as gr
 
 def generate_frames(ffile,dfile): 
-    dataset = FrameReaderDataset(ffile=ffile,dfile=dfile)
+    dataset = FrameDataset(ffile=ffile,dfile=dfile)
     patches = dataset.get_peaks_skimage()
     normalized_patches = dataset.normalize_patches(patches)
     np.save("normalized_patches.npy",normalized_patches)