Deepak107 commited on
Commit
fe906e2
1 Parent(s): 0e32c9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from tensorflow import keras
2
  import gradio as gr
3
- model = keras.models.load_model('potatoes2.h5')
4
  class_names = ['Beer Bottles','Plastic Bottles','Soda Bottle','Water Bottle','Wine Bottle']
5
 
6
  def predict_input_image(img):
 
1
  from tensorflow import keras
2
  import gradio as gr
3
+ model = keras.models.load_model('Bottle_images.h5')
4
  class_names = ['Beer Bottles','Plastic Bottles','Soda Bottle','Water Bottle','Wine Bottle']
5
 
6
  def predict_input_image(img):