Update app.py

app.py

@@ -44,7 +44,7 @@ def generate_code(
         url=url, json={"data": [gen_prompt, max_new_tokens, temperature, seed]}
     )
     generated_text = r.json()["data"][0]
-    generations.append(generated_text)
+    generations.append({model_name: generated_text})
 
 
 def generate_code_threads(
@@ -199,7 +199,7 @@ if st.button("Generate code!", key=4):
         )
         for i in range(len(generations)):
             st.markdown(f"**{selected_models[i]}**")
-            st.code(generations[i])
+            st.code(generations[selected_models[i]])
         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)