Loubna ben allal commited on
Commit
1c7be52
1 Parent(s): 7cf1a13

update app

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- tokenizer = load_tokenizer("lvwerra/codeparrot")
37
- model = 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,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=model, tokenizer=tokenizer)
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")