Update app.py

app.py

@@ -28,6 +28,5 @@ if file is not None:
     prediction = model.predict(image)
     predicted_class = np.argmax(prediction)
 
-    class_names = ['Not Cancer', 'Cancer']
-    st.write(class_names[predicted_class])
+    st.write([predicted_class])