Spaces:

Sa-m
/

Brand-Logo-Classification

Runtime error

Sa-m commited on Sep 10, 2022

Commit

a077d32

•

1 Parent(s): 5e4e204

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ NUM_CLASSES=6
 def classify_image(inp):
   inp = inp.reshape((-1, HEIGHT,WIDTH, 3))
-  inp = tf.keras.applications.nasnet.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, HEIGHT,WIDTH, 3))
+  #inp = tf.keras.applications.nasnet.preprocess_input(inp)
   prediction = model.predict(inp).flatten()
   return {labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}