Update app.py

app.py

@@ -4,10 +4,11 @@ from PIL import Image
 import urllib.request
 import io
 from utils import *
+import tensorflow as tf
 
 # Initialize labels and model
 labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
-model = model_arc() # Assuming this function initializes and returns a trained model
+model = tf.keras.model.load_model('classify_model.h5')
 
 
 # Streamlit UI