update

Dockerfile

@@ -8,8 +8,7 @@ ENV HF_HOME=/code/.cache/huggingface
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
 COPY . .
-
+RUN git clone https://huggingface.co/MissingBreath/recycle-garbage-model
 EXPOSE 5000
 CMD ["python", "api.py"]

api.py

@@ -7,7 +7,8 @@ import io
 app = Flask(__name__)
 
 # Load the model
-model = from_pretrained_keras("MissingBreath/recycle-garbage-model")
+# model = from_pretrained_keras("MissingBreath/recycle-garbage-model")
+model = from_pretrained_keras("./recycle-garbage-model")
 
 # Class labels
 # class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']