Sa-m commited on
Commit
303a695
1 Parent(s): c47223a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import tensorflow as tf
4
  import numpy as np
5
  from tensorflow.keras.models import load_model
6
 
7
- model=load_model('/content/saved_model/best_model.h5')
8
 
9
  def classify_image(inp):
10
  inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
@@ -15,6 +15,6 @@ def classify_image(inp):
15
  image = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE),label='Input')
16
  label = gr.outputs.Label(num_top_classes=2)
17
 
18
- gr.Interface(fn=classify_image, inputs=image, outputs=label, title='Cats Vs Dogs',height=600, width=1200).launch(debug=False)
19
 
20
 
 
4
  import numpy as np
5
  from tensorflow.keras.models import load_model
6
 
7
+ model=load_model('best_model_76.h5')
8
 
9
  def classify_image(inp):
10
  inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
 
15
  image = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE),label='Input')
16
  label = gr.outputs.Label(num_top_classes=2)
17
 
18
+ gr.Interface(fn=classify_image, inputs=image, outputs=label, title='Brand Logo Detection',height=600, width=1200).launch(debug=False)
19
 
20