update app

app.py

@@ -42,18 +42,15 @@ model3 = load_model("facebook/opt-1.3b")
 pipelines = {}
 for model in models:
     if model == "CodeParrot":
-        pipe = pipeline("text-generation", model=model1, tokenizer=tokenizer1)
-        pipelines[model] = pipe
+        pipelines[model] = pipeline("text-generation", model=model1, tokenizer=tokenizer1)
     elif model == "InCoder":
         tokenizer = load_tokenizer("facebook/incoder-1B")
         model = load_model("facebook/incoder-1B")
-        pipe = pipeline("text-generation", model=model2, tokenizer=tokenizer2)
-        pipelines[model] = pipe
+        pipelines[model] = pipeline("text-generation", model=model2, tokenizer=tokenizer2)
     else:
         tokenizer = load_tokenizer("facebook/opt-1.3b")
         model = load_model("facebook/opt-1.3b")
-        pipe = pipeline("text-generation", model=model3, tokenizer=tokenizer3)
-        pipelines[model] = pipe
+        pipelines[model] = pipeline("text-generation", model=model3, tokenizer=tokenizer3)
         
 examples = load_examples()
 example_names = [example["name"] for example in examples]
@@ -97,7 +94,8 @@ elif selected_task == "Code generation":
     if st.button("Generate code!"):
         with st.spinner("Generating code..."):
             for model in selected_models:
+                st.markdown(f"{men(pipelines)} model is {model} keys {pipelines.keys()}:")
                 pipe = pipelines[model]
                 generated_text = pipe(gen_prompt, **gen_kwargs)[0]['generated_text']
-                st.markdown(f"### {model}:")
+                st.markdown(f"{model}:")
                 st.code(generated_text)