add multiprocessing

app.py

@@ -19,8 +19,7 @@ def generate_code(model_name, gen_prompt, max_new_tokens, temperature, seed):
     url = f'https://hf.space/embed/loubnabnl/{model_name.lower()}-subspace/+/api/predict/'
     r = requests.post(url=url, json={"data": [gen_prompt, max_new_tokens, temperature, seed]})
     generated_text = r.json()['data'][0]
-    st.markdown(model_name)
-    st.code(generated_text)
+    return generated_text
     
 st.set_page_config(page_icon=":laptop:", layout="wide")
 
@@ -89,4 +88,7 @@ elif selected_task == "Code generation":
                            max_new_tokens=max_new_tokens, 
                            temperature=temperature,
                            seed=seed)
-            pool.map(generate_parallel, selected_models)
+            output = pool.map(generate_parallel, selected_models)
+            for i in range(len(output)):
+                st.markdown(selected_models[i])
+                st.code(output[i])