Raman Dutt
commited on
Commit
•
3a33055
1
Parent(s):
4cfadd6
minor
Browse files
app.py
CHANGED
|
@@ -138,6 +138,7 @@ def predict(
|
|
| 138 |
)
|
| 139 |
|
| 140 |
result_pil_image = result_image["images"][0]
|
|
|
|
| 141 |
|
| 142 |
# Create a Bar Plot displaying the number of tunable parameters for the selected PEFT Type
|
| 143 |
df = pd.DataFrame(
|
|
@@ -147,6 +148,8 @@ def predict(
|
|
| 147 |
}
|
| 148 |
)
|
| 149 |
|
|
|
|
|
|
|
| 150 |
df = df[df["Fine-Tuning Strategy"].isin(["full", unet_pretraining_type])].reset_index(
|
| 151 |
drop=True
|
| 152 |
)
|
|
|
|
| 138 |
)
|
| 139 |
|
| 140 |
result_pil_image = result_image["images"][0]
|
| 141 |
+
|
| 142 |
|
| 143 |
# Create a Bar Plot displaying the number of tunable parameters for the selected PEFT Type
|
| 144 |
df = pd.DataFrame(
|
|
|
|
| 148 |
}
|
| 149 |
)
|
| 150 |
|
| 151 |
+
print(df)
|
| 152 |
+
|
| 153 |
df = df[df["Fine-Tuning Strategy"].isin(["full", unet_pretraining_type])].reset_index(
|
| 154 |
drop=True
|
| 155 |
)
|