Update app.py

app.py

@@ -1,6 +1,7 @@
 import h5py
 import gradio as gr
 import scipy.io as io
+import matplotlib.pyplot as plt
 import PIL.Image as Image
 import numpy as np
 from torchvision import transforms