trying gdown

Dockerfile

@@ -9,6 +9,10 @@ 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
+# RUN git clone https://huggingface.co/MissingBreath/recycle-garbage-model
+
+# RUN pip install gdown
+RUN gdown 1-2xLqBeuq6ypWyPEVdfslKU1GLgD9a1_
+
 EXPOSE 5000
 CMD ["python", "api.py"]

api.py

@@ -3,13 +3,14 @@ from huggingface_hub import from_pretrained_keras
 import numpy as np
 from PIL import Image
 import io
+import tensorflow as tf
 
 app = Flask(__name__)
 
 # Load the model
 # model = from_pretrained_keras("MissingBreath/recycle-garbage-model")
-model = from_pretrained_keras("./recycle-garbage-model")
-
+# model = from_pretrained_keras("./recycle-garbage-model")
+model = tf.keras.models.load_model('_9217')
 # Class labels
 # class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
 

requirements.txt

@@ -3,4 +3,5 @@ numpy
 pillow
 huggingface_hub
 tensorflow==2.15.0
-keras==2.15.0
+keras==2.15.0
+gdown==5.1.0