Update app.py

app.py

@@ -199,7 +199,9 @@ if st.button("Generate code!", key=4):
         )
         for i in range(len(generations)):
             st.markdown(f"**{selected_models[i]}**")
-            st.code(generations[selected_models[i]])
+            for i in range(len(generations)):
+                if selected_models in generations[i].keys():
+                    st.code(generations[i][selected_models])
         if len(generations) < len(selected_models):
             st.markdown("<span style='color:red'>Warning: Some models run into timeout, you can try generating code using the original subspaces: [InCoder](https://huggingface.co/spaces/loubnabnl/incoder-subspace), [CodeGen](https://huggingface.co/spaces/loubnabnl/codegen-subspace), [CodeParrot](https://huggingface.co/spaces/loubnabnl/codeparrot-subspace)</span>", unsafe_allow_html=True)