Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -131,7 +131,7 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
|
|
131 |
# Add other hyperparameters as needed
|
132 |
}
|
133 |
# The base model you will train a LoRA on top of
|
134 |
-
base_model_path = "facebook/esm2_t12_35M_UR50D"
|
135 |
|
136 |
# Define labels and model
|
137 |
#id2label = {0: "No binding site", 1: "Binding site"}
|
|
|
131 |
# Add other hyperparameters as needed
|
132 |
}
|
133 |
# The base model you will train a LoRA on top of
|
134 |
+
#base_model_path = "facebook/esm2_t12_35M_UR50D"
|
135 |
|
136 |
# Define labels and model
|
137 |
#id2label = {0: "No binding site", 1: "Binding site"}
|