PBJ commited on
Commit
e3fe776
·
1 Parent(s): bca8baf

Update Welcome.py

Browse files
Files changed (1) hide show
  1. Welcome.py +1 -1
Welcome.py CHANGED
@@ -76,7 +76,7 @@ if f is not None:
76
  test_df = tf.convert_to_tensor(test_df)
77
 
78
  # Inference
79
- interpreter = tf.lite.Interpreter("model.tflite")
80
  prediction_fn = interpreter.get_signature_runner("serving_default")
81
  output = prediction_fn(inputs=test_df)
82
  sign = np.argmax(output["outputs"])
 
76
  test_df = tf.convert_to_tensor(test_df)
77
 
78
  # Inference
79
+ interpreter = tf.lite.Interpreter("models/model.tflite")
80
  prediction_fn = interpreter.get_signature_runner("serving_default")
81
  output = prediction_fn(inputs=test_df)
82
  sign = np.argmax(output["outputs"])