Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ pil_to_tensor = transforms.ToTensor()
|
|
14 |
# Define a list of available weight files
|
15 |
weights_files = {
|
16 |
"flolv2_UHDLL": './weights/flolv2_UHDLL.pt',
|
17 |
-
"
|
18 |
}
|
19 |
|
20 |
# Initial model setup (without weights)
|
|
|
14 |
# Define a list of available weight files
|
15 |
weights_files = {
|
16 |
"flolv2_UHDLL": './weights/flolv2_UHDLL.pt',
|
17 |
+
"flolv2_LOLv2-Real": './weights/flolv2_all_111439.pt', # Add other available weights here
|
18 |
}
|
19 |
|
20 |
# Initial model setup (without weights)
|