ayaanzaveri commited on
Commit
c959fe1
·
1 Parent(s): 6de32fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import tensorflow as tf
2
  import numpy as np
 
3
  import gradio as gr
4
 
 
 
5
  model = tf.keras.models.load_model("mnist-model.h5")
6
 
7
  def recognize_digit(image):
 
1
  import tensorflow as tf
2
  import numpy as np
3
+ from urllib.request import urlretrieve
4
  import gradio as gr
5
 
6
+ urlretrieve("https://github.com/AyaanZaveri/mnist/raw/main/mnist-model.h5", "mnist-model.h5")
7
+
8
  model = tf.keras.models.load_model("mnist-model.h5")
9
 
10
  def recognize_digit(image):