Update app.py

app.py

@@ -71,15 +71,20 @@ def image_movement(input1, input2):
 def get_sample_images(num_images):
     sample_images = []
     for i in range(num_images):
-      test_feature = x_test[i]
+      test_feature = X_test[i]
       test_feature_2d =test_feature.reshape(28,28)
     
       # Make it unsigned integers:
       data = test_feature_2d.astype(np.uint8)
 
-      sample_images.append([data,int(np.random.choice(['shift up', 'shift down', 'shift left', 'shift right', 'rotate left', 'rotate right']))])   # ["image path", "K"]
+      outdir =  "images_folder"
+      img_path = os.path.join(outdir, 'local_%05d.png' % (i,))
+      if not os.path.exists(outdir):
+         os.mkdir(outdir)
+      cv2.imwrite(img_path, data)
+
+      sample_images.append([img_path,np.random.choice(['shift up', 'shift down', 'shift left', 'shift right', 'rotate left', 'rotate right'])])   # ["image path", "K"]
     return sample_images
-    
 sample_images = get_sample_images(10)