Update app.py

app.py

@@ -8,7 +8,8 @@ import matplotlib.pyplot as plt
 
 def get_digit(digit_choice):
   digit = train_images[digit_choice]
-  fig = plt.imshow(digit)
+  fig = plt.figure(figsize=(10, 10))
+  plt.imshow(digit)
   out_txt = "digit: %d" % digit_choice
   return fig, out_txt