jykoh commited on
Commit
8c18853
1 Parent(s): 9d41210

Fix queue comment

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. fromage/models.py +1 -1
app.py CHANGED
@@ -212,5 +212,5 @@ with gr.Blocks(css=css) as demo:
212
  save_button.click(None, [], [], _js=save_js)
213
 
214
 
215
- # demo.queue(concurrency_count=1, api_open=False, max_size=16)
216
- demo.launch(debug=True, server_name="127.0.0.1")
 
212
  save_button.click(None, [], [], _js=save_js)
213
 
214
 
215
+ demo.queue(concurrency_count=1, api_open=False, max_size=16)
216
+ demo.launch(debug=True, server_name="0.0.0.0")
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 = True
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'