juaben commited on
Commit
59b43a3
·
verified ·
1 Parent(s): c079982

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- "flolv2_other": './weights/flolv2_all_111439.pt', # Add other available weights here
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)