Raman Dutt
commited on
Commit
•
4cfadd6
1
Parent(s):
d530147
minor fix
Browse files
app.py
CHANGED
|
@@ -147,7 +147,7 @@ def predict(
|
|
| 147 |
}
|
| 148 |
)
|
| 149 |
|
| 150 |
-
df = df[df["
|
| 151 |
drop=True
|
| 152 |
)
|
| 153 |
|
|
|
|
| 147 |
}
|
| 148 |
)
|
| 149 |
|
| 150 |
+
df = df[df["Fine-Tuning Strategy"].isin(["full", unet_pretraining_type])].reset_index(
|
| 151 |
drop=True
|
| 152 |
)
|
| 153 |
|