reformat code

app.py

@@ -8,6 +8,7 @@ import streamlit as st
 MODELS = ["CodeParrot", "InCoder", "CodeGen", "PolyCoder"]
 GENERATION_MODELS = ["CodeParrot", "InCoder", "CodeGen"]
 
+
 @st.cache()
 def load_examples():
     with open("utils/examples.json", "r") as f:
@@ -18,10 +19,12 @@ def load_examples():
 def read_markdown(path):
     with open(path, "r") as f:
         output = f.read()
-    st.markdown(output, unsafe_allow_html=True)   
-    
-    
-def generate_code(generations, model_name, gen_prompt, max_new_tokens, temperature, seed):
+    st.markdown(output, unsafe_allow_html=True)
+
+
+def generate_code(
+    generations, model_name, gen_prompt, max_new_tokens, temperature, seed
+):
     # call space using its API endpoint
     url = (
         f"https://hf.space/embed/loubnabnl/{model_name.lower()}-subspace/+/api/predict/"
@@ -30,21 +33,33 @@ def generate_code(generations, model_name, gen_prompt, max_new_tokens, temperatu
         url=url, json={"data": [gen_prompt, max_new_tokens, temperature, seed]}
     )
     generated_text = r.json()["data"][0]
-    generations.append(generated_text)    
-
-        
-def generate_code_threads(generations, models, gen_prompt, max_new_tokens, temperature, seed):                                                
-    threads = []                                                                
-    for model_name in models:                                                     
-        # create the thread                                                     
-        threads.append(                                                         
-            threading.Thread(target=generate_code, args=(generations, model_name, gen_prompt, max_new_tokens, temperature, seed))
-        )         
-        threads[-1].start()                 
-                                        
-    for t in threads:                                                           
-        t.join()  
-        
+    generations.append(generated_text)
+
+
+def generate_code_threads(
+    generations, models, gen_prompt, max_new_tokens, temperature, seed
+):
+    threads = []
+    for model_name in models:
+        # create the thread
+        threads.append(
+            threading.Thread(
+                target=generate_code,
+                args=(
+                    generations,
+                    model_name,
+                    gen_prompt,
+                    max_new_tokens,
+                    temperature,
+                    seed,
+                ),
+            )
+        )
+        threads[-1].start()
+
+    for t in threads:
+        t.join()
+
 
 st.set_page_config(page_icon=":laptop:", layout="wide")
 with open("utils/table_contents.txt", "r") as f:
@@ -59,12 +74,12 @@ read_markdown("utils/intro.txt")
 st.subheader("1 - Code datasets")
 read_markdown("datasets/intro.txt")
 read_markdown("datasets/github_code.txt")
-#GITHUB_CODE = "https://huggingface.co/datasets/lvwerra/github-code"
-#st.markdown(f"Preview of some code files from Github repositories in [Github-code dataset]({GITHUB_CODE}):")
-#df = pd.read_csv("utils/data_preview.csv")
-#st.dataframe(df)
+# GITHUB_CODE = "https://huggingface.co/datasets/lvwerra/github-code"
+# st.markdown(f"Preview of some code files from Github repositories in [Github-code dataset]({GITHUB_CODE}):")
+# df = pd.read_csv("utils/data_preview.csv")
+# st.dataframe(df)
 
-col1, col2= st.columns([1,2])
+col1, col2 = st.columns([1, 2])
 with col1:
     selected_model = st.selectbox("", MODELS, key=1)
 read_markdown(f"datasets/{selected_model.lower()}.txt")
@@ -73,7 +88,7 @@ read_markdown(f"datasets/{selected_model.lower()}.txt")
 # Model architecture
 st.subheader("2 - Model architecture")
 read_markdown("architectures/intro.txt")
-col1, col2= st.columns([1,2])
+col1, col2 = st.columns([1, 2])
 with col1:
     selected_model = st.selectbox("", MODELS, key=2)
 read_markdown(f"architectures/{selected_model.lower()}.txt")
@@ -85,12 +100,15 @@ read_markdown("evaluation/demo_humaneval.txt")
 
 # Code generation
 st.subheader("4 - Code generation ✨")
-col1, col2, col3 = st.columns([7,1,6])
+col1, col2, col3 = st.columns([7, 1, 6])
 with col1:
     st.markdown("**Models**")
     selected_models = st.multiselect(
-    "Select code generation models to compare:", GENERATION_MODELS, default=["CodeParrot"], key=3
-)
+        "Select code generation models to compare:",
+        GENERATION_MODELS,
+        default=["CodeParrot"],
+        key=3,
+    )
     st.markdown(" ")
     st.markdown("**Examples**")
     examples = load_examples()
@@ -113,9 +131,7 @@ with col3:
         step=4,
         max_value=256,
     )
-    seed = st.slider(
-        "Random seed:", value=42, min_value=0, step=1, max_value=1000
-    )
+    seed = st.slider("Random seed:", value=42, min_value=0, step=1, max_value=1000)
 gen_prompt = st.text_area(
     "Generate code with prompt:",
     value=example_text,
@@ -141,4 +157,4 @@ if st.button("Generate code!"):
 
 # Resources
 st.subheader("Resources")
-read_markdown("utils/resources.txt")
+read_markdown("utils/resources.txt")