Spaces:
Build error
Build error
Fix deployment of debug model
Browse files- app.py +3 -3
- fromage/models.py +1 -1
app.py
CHANGED
@@ -208,6 +208,6 @@ with gr.Blocks(css=css) as demo:
|
|
208 |
save_button.click(None, [], [], _js=save_js)
|
209 |
|
210 |
|
211 |
-
|
212 |
-
|
213 |
-
demo.launch(debug=True, server_name="127.0.0.1")
|
|
|
208 |
save_button.click(None, [], [], _js=save_js)
|
209 |
|
210 |
|
211 |
+
demo.queue(concurrency_count=1, api_open=False, max_size=16)
|
212 |
+
demo.launch(debug=True, server_name="0.0.0.0")
|
213 |
+
# demo.launch(debug=True, server_name="127.0.0.1")
|
fromage/models.py
CHANGED
@@ -640,7 +640,7 @@ def load_fromage(embeddings_dir: str, model_args_path: str, model_ckpt_path: str
|
|
640 |
assert len(ret_token_idx) == 1, ret_token_idx
|
641 |
model_kwargs['retrieval_token_idx'] = ret_token_idx[0]
|
642 |
|
643 |
-
debug =
|
644 |
if debug:
|
645 |
model_kwargs['opt_version'] = 'facebook/opt-125m'
|
646 |
model_kwargs['visual_encoder'] = 'openai/clip-vit-base-patch32'
|
|
|
640 |
assert len(ret_token_idx) == 1, ret_token_idx
|
641 |
model_kwargs['retrieval_token_idx'] = ret_token_idx[0]
|
642 |
|
643 |
+
debug = False
|
644 |
if debug:
|
645 |
model_kwargs['opt_version'] = 'facebook/opt-125m'
|
646 |
model_kwargs['visual_encoder'] = 'openai/clip-vit-base-patch32'
|