Update app.py

app.py

@@ -34,12 +34,12 @@ def upload_predict(upload_image, model):
 
         # Predicts the model
         prediction = model.predict(image)
-       
-    
-        
-        pred_class=decode_predictions(prediction,top=1)
+        index = np.argmax(prediction)
+        classname = ['Glass', 'NoGlass']
+        class_name = classname[index]
+
         
-        return pred_class
+        return class_name
 if file is None:
     st.text("Please upload an image file")
 else: