Update app.py

app.py

@@ -34,7 +34,7 @@ def infer(input_image):
     image_tensor.set_shape([None, None, 3])
     image_tensor = tf.image.resize(image_tensor, (IMG_SIZE, IMG_SIZE))
     print(image_tensor.shape)
-    predictions = teacher_model.predict(np.expand_dims((image_tensor), axis=0))
+    predictions = student_model.predict(np.expand_dims((image_tensor), axis=0))
     print(predictions)
     predictions = np.squeeze(predictions)
     print(predictions)