Update app.py

app.py

@@ -6,5 +6,5 @@ model = load_model('tox_model.h5')
 text = st.text_area('Enter some text')
 
 if text:
-  out = model.predict(text)
+  out[6] = model.predict(text)
   st.json(out)