tonyliu404 commited on
Commit
2a9e1bc
·
verified ·
1 Parent(s): 41fd686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -237,7 +237,7 @@ def classifyImage(input_image):
237
 
238
  predictions = model.predict(input_array)[0]
239
  print(f"Predictions: {predictions}")
240
- predictions = tf.nn.softmax(predictions).numpy() #TESTING
241
  print(f"Predictions AFTER SOFTMAX: {predictions}")
242
 
243
  probability_sum = predictions.sum() * 100
 
237
 
238
  predictions = model.predict(input_array)[0]
239
  print(f"Predictions: {predictions}")
240
+ predictions = softmax_with_temperature(predictions, temperature=2.0).numpy() #testing
241
  print(f"Predictions AFTER SOFTMAX: {predictions}")
242
 
243
  probability_sum = predictions.sum() * 100