Spaces:
Build error
Build error
load pipeline fix
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ st.set_page_config(page_icon=':parrot:', layout="wide")
|
|
21 |
device = 1 if torch.cuda.is_available() else 0
|
22 |
device_name = "GPU" if torch.cuda.is_available() else "CPU"
|
23 |
model_ckpt = "lvwerra/codeparrot"
|
24 |
-
pipe =
|
25 |
examples = load_examples()
|
26 |
example_names = [example["name"] for example in examples]
|
27 |
name2id = dict([(name, i) for i, name in enumerate(example_names)])
|
|
|
21 |
device = 1 if torch.cuda.is_available() else 0
|
22 |
device_name = "GPU" if torch.cuda.is_available() else "CPU"
|
23 |
model_ckpt = "lvwerra/codeparrot"
|
24 |
+
pipe = load_pipeline(model_ckpt, device)
|
25 |
examples = load_examples()
|
26 |
example_names = [example["name"] for example in examples]
|
27 |
name2id = dict([(name, i) for i, name in enumerate(example_names)])
|