Spaces:
Sleeping
Sleeping
fix: handle empty string for model revision
Browse files
app.py
CHANGED
@@ -92,6 +92,10 @@ def translate_point_coords_from_out_to_in(point=None, input_image_size=None, out
|
|
92 |
def process_refexp(image: Image, prompt: str, model_revision: str = 'main'):
|
93 |
|
94 |
print(f"(image, prompt): {image}, {prompt}")
|
|
|
|
|
|
|
|
|
95 |
print(f"model checkpoint revision: {model_revision}")
|
96 |
|
97 |
load_model(model_revision)
|
|
|
92 |
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 |
|
101 |
load_model(model_revision)
|