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

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_other.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_other": './weights/flolv2_all_111439.pt', # Add other available weights here
18
  }
19
 
20
  # Initial model setup (without weights)