Spaces:

sharktide
/

recycleai-api

Sleeping

sharktide commited on Jan 2

Commit

a80f51a

1 Parent(s): e10a731

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from io import BytesIO
 from fastapi.responses import JSONResponse
 # Load your trained model (make sure it's available in the working directory)
-model = tf.keras.models.load_model('model.h5')
 # Class names for predictions (modify if necessary)
 class_names = ['Glass', 'Metal', 'Paperboard', 'Plastic-Polystyrene', 'Plastic-Regular']

 from fastapi.responses import JSONResponse
 # Load your trained model (make sure it's available in the working directory)
+model = tf.keras.models.load_model('recyclebot.keras')
 # Class names for predictions (modify if necessary)
 class_names = ['Glass', 'Metal', 'Paperboard', 'Plastic-Polystyrene', 'Plastic-Regular']