Stefaron commited on
Commit
12f297e
·
verified ·
1 Parent(s): 2c50213

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,5 +31,5 @@ This model is a fine-tuned ResNet50 for classifying trash images into categories
31
  ## How to Use
32
  Load the model and use it for inference:
33
  ```python
34
- from transformers import AutoModel
35
- model = AutoModel.from_pretrained("Stefaron/trash-classifier")
 
31
  ## How to Use
32
  Load the model and use it for inference:
33
  ```python
34
+ import tensorflow as tf
35
+ model = tf.keras.models.load_model("Stefaron/trash-classifier/best_model.keras")