dk00069 commited on
Commit
bb9892d
·
verified ·
1 Parent(s): bc1423c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import gradio as gr
2
  import numpy as np
3
- from tensorflow import keras
4
  from PIL import Image
5
 
6
- model = keras.models.load_model("waste_classifier_final.h5")
 
 
 
7
 
8
  labels = [
9
  "Cardboard",
 
1
  import gradio as gr
2
  import numpy as np
3
+ import tensorflow as tf
4
  from PIL import Image
5
 
6
+ model = tf.keras.models.load_model(
7
+ "waste_classifier_final.h5",
8
+ compile=False
9
+ )
10
 
11
  labels = [
12
  "Cardboard",