hwberry2 commited on
Commit
70ae181
·
1 Parent(s): 17043a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,9 +40,9 @@ with gr.Blocks() as demo:
40
 
41
 
42
  model.fit(x_train, y_train, epochs=5)
43
- model_eval_data = model.evaluate(x_test, y_test, verbose=2)
44
 
45
- return model_eval_data
46
 
47
 
48
  # Creates the Gradio interface objects
 
40
 
41
 
42
  model.fit(x_train, y_train, epochs=5)
43
+ test_loss, test_acc = model.evaluate(x_test, y_test, verbose=2)
44
 
45
+ return('\nTest accuracy:', test_acc)
46
 
47
 
48
  # Creates the Gradio interface objects