UI changes

app.py

@@ -37,17 +37,13 @@ css = """
         background-image: linear-gradient(to right bottom, rgba(130,217,244, 0.9), rgba(158,231,214, 1.0));
     }
     #save-btn:hover {
-        background-image: linear-gradient(to right bottom, rgba(130,217,244, 0.9), rgba(158,231,214, 1.0));
+        background-image: linear-gradient(to right bottom, rgba(110,197,224, 0.9), rgba(138,211,194, 1.0));
     }
     #share-btn-2 {
         background-image: linear-gradient(to right bottom, rgba(130,217,244, 0.9), rgba(158,231,214, 1.0));
     }
     #share-btn-2:hover {
-        background-image: linear-gradient(to right bottom, rgba(130,217,244, 0.9), rgba(158,231,214, 1.0));
-    }
-    .message .user {
-    }
-    .message .bot {
+        background-image: linear-gradient(to right bottom, rgba(110,197,224, 0.9), rgba(138,211,194, 1.0));
     }
 """
 
@@ -166,7 +162,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column(scale=0.7, min_width=500):
             with gr.Row():
-                chatbot = gr.Chatbot(elem_id="chatbot", label="FROMAGe Chatbot")
+                chatbot = gr.Chatbot(elem_id="chatbot", label="🧀 FROMAGe Chatbot")
             with gr.Row():
                 image_btn = gr.UploadButton("🖼️ Upload Image", file_types=["image"])
 
@@ -190,7 +186,7 @@ with gr.Blocks(css=css) as demo:
             max_ret_images = gr.Number(
                 minimum=0, maximum=3, value=2, precision=1, interactive=True, label="Max images to return")
             gr_max_len = gr.Slider(minimum=1, maximum=64, value=32,
-                                   step=1, interactive=True, label="Max # of words returned")
+                                   step=1, interactive=True, label="Max # of words")
             gr_temperature = gr.Slider(
                 minimum=0.0, maximum=1.0, value=0.0, interactive=True, label="Temperature (0 for deterministic, higher for more randomness)")
 
@@ -212,5 +208,6 @@ with gr.Blocks(css=css) as demo:
     save_button.click(None, [], [], _js=save_js)
 
 
-demo.queue(concurrency_count=1, api_open=False, max_size=16)
-demo.launch(debug=True, server_name="0.0.0.0")
+# demo.queue(concurrency_count=1, api_open=False, max_size=16)
+# demo.launch(debug=True, server_name="0.0.0.0")
+demo.launch(debug=True, server_name="127.0.0.1")

fromage/models.py

@@ -640,7 +640,7 @@ def load_fromage(embeddings_dir: str, model_args_path: str, model_ckpt_path: str
   assert len(ret_token_idx) == 1, ret_token_idx
   model_kwargs['retrieval_token_idx'] = ret_token_idx[0]
 
-  debug = False
+  debug = True
   if debug:
     model_kwargs['opt_version'] = 'facebook/opt-125m'
     model_kwargs['visual_encoder'] = 'openai/clip-vit-base-patch32'