Loubna ben allal
commited on
Commit
•
1c7be52
1
Parent(s):
7cf1a13
update app
Browse files
app.py
CHANGED
@@ -33,8 +33,8 @@ tasks = [" ","Model architecture", "Model evaluation", "Pretraining dataset", "C
|
|
33 |
selected_task = st.sidebar.selectbox("Select a task:", tasks)
|
34 |
|
35 |
|
36 |
-
|
37 |
-
|
38 |
tokenizer2 = load_tokenizer("facebook/incoder-1B")
|
39 |
model2 = load_model("facebook/incoder-1B")
|
40 |
tokenizer3 = load_tokenizer("facebook/opt-1.3b")
|
@@ -42,7 +42,7 @@ model3 = load_model("facebook/opt-1.3b")
|
|
42 |
pipelines = {}
|
43 |
for model in models:
|
44 |
if model == "CodeParrot":
|
45 |
-
pipe = pipeline("text-generation", model=
|
46 |
pipelines[model] = pipe
|
47 |
elif model == "InCoder":
|
48 |
tokenizer = load_tokenizer("facebook/incoder-1B")
|
|
|
33 |
selected_task = st.sidebar.selectbox("Select a task:", tasks)
|
34 |
|
35 |
|
36 |
+
tokenizer1 = load_tokenizer("lvwerra/codeparrot")
|
37 |
+
model1 = load_model("lvwerra/codeparrot")
|
38 |
tokenizer2 = load_tokenizer("facebook/incoder-1B")
|
39 |
model2 = load_model("facebook/incoder-1B")
|
40 |
tokenizer3 = load_tokenizer("facebook/opt-1.3b")
|
|
|
42 |
pipelines = {}
|
43 |
for model in models:
|
44 |
if model == "CodeParrot":
|
45 |
+
pipe = pipeline("text-generation", model=model1, tokenizer=tokenizer1)
|
46 |
pipelines[model] = pipe
|
47 |
elif model == "InCoder":
|
48 |
tokenizer = load_tokenizer("facebook/incoder-1B")
|