tt

api.py

@@ -5,15 +5,12 @@ from PIL import Image
 import io
 import tensorflow as tf
 
-app = FastAPI()
 
-# Load the model
-# model = from_pretrained_keras("MissingBreath/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']
 
+app = FastAPI()
 @app.post("/classify")
 async def classify(image: UploadFile = File(...)):
     if image is not None:
@@ -23,6 +20,7 @@ async def classify(image: UploadFile = File(...)):
         img_array = np.expand_dims(img_array, axis=0)
         predictions = model.predict(img_array)
         predicted_class_idx = np.argmax(predictions)
+        predicted_class_idx = int(predicted_class_idx)  
         # predicted_class = class_labels[predicted_class_idx]
         # return {"prediction": predicted_class}
         return {"prediction": predicted_class_idx}