Spaces:

Sa-m
/

Brand-Logo-Classification

Runtime error

Sa-m commited on Jun 25, 2022

Commit

446f348

1 Parent(s): d780d68

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model=load_model('best_model_76.h5')
 def classify_image(inp):
   inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
-  inp = tf.keras.applications.vgg16.preprocess_input(inp)
   prediction = model.predict(inp).flatten()
   return {labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}

 def classify_image(inp):
   inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
+  #inp = tf.keras.applications.vgg16.preprocess_input(inp)
   prediction = model.predict(inp).flatten()
   return {labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}