ivelin commited on
Commit
ccf9890
·
1 Parent(s): 9cdce98

fix: model revision handling

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -93,8 +93,8 @@ def process_refexp(image: Image, prompt: str, model_revision: str = 'main'):
93
 
94
  print(f"(image, prompt): {image}, {prompt}")
95
 
96
- if not pretrained_revision:
97
- pretrained_revision = 'main'
98
 
99
  print(f"model checkpoint revision: {model_revision}")
100
 
 
93
 
94
  print(f"(image, prompt): {image}, {prompt}")
95
 
96
+ if not model_revision:
97
+ model_revision = 'main'
98
 
99
  print(f"model checkpoint revision: {model_revision}")
100