Spaces:
Runtime error
Runtime error
sabre-code
commited on
Commit
•
6f98250
1
Parent(s):
ca01b1d
Update classifier.py
Browse files- classifier.py +2 -2
classifier.py
CHANGED
@@ -3,8 +3,8 @@ from tf.keras.models import load_model
|
|
3 |
model = load_model('myModel.hdf5')
|
4 |
|
5 |
import gradio as gr
|
6 |
-
|
7 |
-
def classify(
|
8 |
img_array = tf.keras.utils.img_to_array(img)
|
9 |
img_array = tf.expand_dims(img_array, 0) # Create a batch
|
10 |
predictions = model.predict(img_array)
|
|
|
3 |
model = load_model('myModel.hdf5')
|
4 |
|
5 |
import gradio as gr
|
6 |
+
from gradio import inputs
|
7 |
+
def classify(img):
|
8 |
img_array = tf.keras.utils.img_to_array(img)
|
9 |
img_array = tf.expand_dims(img_array, 0) # Create a batch
|
10 |
predictions = model.predict(img_array)
|