Update app.py

app.py

@@ -62,7 +62,7 @@ def loading_data(img):
 
 def predict(img):
     if img is None:
-        return "No image selected", plt.figure(np.ones((30,30)))
+        return "No image selected", plt.figure()
     """the main process of inference"""
     test_loader = loading_data(img)
     #model = SASNet()