Update app.py

app.py

@@ -65,5 +65,8 @@ if type(mol) == None:
     scores.append(None)
 output += scores
 output = [input_compound] + output
-output_df = pd.DataFrame(np.array(output).reshape(1, -1), columns=['input'] + [f'{i}th' for i in range(CFG.num_beams)] + ['valid compound'] + [f'{i}th score' for i in range(CFG.num_beams)] + ['valid compound score'])
-st.table(output_df)
+try:
+    output_df = pd.DataFrame(np.array(output).reshape(1, -1), columns=['input'] + [f'{i}th' for i in range(CFG.num_beams)] + ['valid compound'] + [f'{i}th score' for i in range(CFG.num_beams)] + ['valid compound score'])
+    st.table(output_df)
+except:
+    pass