Spaces:
Running
Running
Commit ·
6aa9fe5
1
Parent(s): ac8676d
Fixed None token to get_pipeline function
Browse files
protac_splitter/protac_splitter.py
CHANGED
|
@@ -134,6 +134,7 @@ def split_protac(
|
|
| 134 |
if use_transformer:
|
| 135 |
pipe = get_pipeline(
|
| 136 |
model_name="ailab-bio/PROTAC-Splitter",
|
|
|
|
| 137 |
is_causal_language_model=False,
|
| 138 |
num_return_sequences=beam_size,
|
| 139 |
device=device,
|
|
|
|
| 134 |
if use_transformer:
|
| 135 |
pipe = get_pipeline(
|
| 136 |
model_name="ailab-bio/PROTAC-Splitter",
|
| 137 |
+
token=os.environ.get("HF_TOKEN", None),
|
| 138 |
is_causal_language_model=False,
|
| 139 |
num_return_sequences=beam_size,
|
| 140 |
device=device,
|