ZhengPeng7 commited on
Commit
a2fe5a2
·
1 Parent(s): 6dd9966

Add inference endpoint feature in HF model page.

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -75,7 +75,7 @@ usage_to_weights_file = {
75
  }
76
 
77
  # Choose the version of BiRefNet here.
78
- usage = 'General'
79
 
80
  # Set resolution
81
  if usage in ['General-Lite-2K']:
 
75
  }
76
 
77
  # Choose the version of BiRefNet here.
78
+ usage = 'General-legacy'
79
 
80
  # Set resolution
81
  if usage in ['General-Lite-2K']: