.

app.py

@@ -141,13 +141,24 @@ st.write(mono)
 #st.write(MessagePassingModel)
 #st.write(test_weights)
 #
+st.write(atoms)
+st.write(dcmol)
 
 from ase.visualize import view
-display_mol = view(atoms+dcmol, viewer="x3d")
 
-st.write(type(display_mol))
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
+
+output = StringIO()
+display_mol = (atoms+dcmol).write(output, format="html")
+data = display_mol.getvalue()
+
+st.write(type(data))
+
+st.html(data)
 
-st.html(display_mol)
 
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)