update app

app.py

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