added stuff

app.py

@@ -8,7 +8,7 @@ login(token=os.environ["HUGGINGFACE_TOKEN"])
 
 demo_imgs = [
     ["images/chinchilla_web-1024x683.jpg", "images/shiba-inu-dog-in-the-snow.jpg"],
-    ["images/900.jpeg", "images/hummus.jpg", "images/london-underground-sign.jpg", "images/COCO_train2014_000000194806.jpg"],
+    ["images/900.jpeg", "images/hummus.jpg"],
     ["images/COCO_train2014_000000572279.jpg", "images/COCO_train2014_000000194806.jpg"],
     [
         "images/bcee7a-20190225-a-london-underground-sign.jpg",
@@ -167,7 +167,7 @@ def generate(
 with gr.Blocks() as demo:
     # As a consequence, you should treat this model as a research prototype and not as a production-ready model. Before using this demo please familiarize yourself with our [model card](https://github.com/mlfoundations/open_flamingo/blob/main/MODEL_CARD.md) and [terms and conditions](https://github.com/mlfoundations/open_flamingo/blob/main/TERMS_AND_CONDITIONS.md)
     gr.Markdown(
-    """
+        """
     # 🦩 OpenFlamingo Demo  
     
     Blog posts: #1 [An open-source framework for training vision-language models with in-context learning](https://laion.ai/blog/open-flamingo/) // #2 [OpenFlamingo v2: New Models and Enhanced Training Setup]()\n
@@ -206,10 +206,11 @@ RedPajama-INCITE-Base-3B-v1](https://huggingface.co/togethercomputer/RedPajama-I
                 query_image = gr.Image(type="pil")
                 text_output = gr.Textbox(value="Output:", label="Model output")
 
+        
         run_btn = gr.Button("Run model")
 
-        def on_click_fn(img): return generate(1, img, "", tc=read_tc)
-        run_btn.click(on_click_fn, inputs=[query_image], outputs=[text_output])
+        def on_click_fn(img, read_tc): return generate(1, img, "", tc=read_tc)
+        run_btn.click(on_click_fn, inputs=[query_image, read_tc], outputs=[text_output])
 
     with gr.Tab("🦓 Animal recognition"):
         with gr.Row():
@@ -233,8 +234,8 @@ RedPajama-INCITE-Base-3B-v1](https://huggingface.co/togethercomputer/RedPajama-I
 
         run_btn = gr.Button("Run model")
 
-        def on_click_fn(img): return generate(0, img, "", tc=read_tc)
-        run_btn.click(on_click_fn, inputs=[query_image], outputs=[text_output])
+        def on_click_fn(img, read_tc): return generate(0, img, "", tc=read_tc)
+        run_btn.click(on_click_fn, inputs=[query_image, read_tc], outputs=[text_output])
 
     with gr.Tab("🔢 Counting objects"):
         with gr.Row():
@@ -258,8 +259,8 @@ RedPajama-INCITE-Base-3B-v1](https://huggingface.co/togethercomputer/RedPajama-I
 
         run_btn = gr.Button("Run model")
 
-        def on_click_fn(img): return generate(4, img, "", tc=read_tc)
-        run_btn.click(on_click_fn, inputs=[query_image], outputs=[text_output])
+        def on_click_fn(img, read_tc): return generate(4, img, "", tc=read_tc)
+        run_btn.click(on_click_fn, inputs=[query_image, read_tc], outputs=[text_output])
 
     with gr.Tab("🕵️ Visual Question Answering"):
         with gr.Row():
@@ -285,9 +286,9 @@ RedPajama-INCITE-Base-3B-v1](https://huggingface.co/togethercomputer/RedPajama-I
                 text_output = gr.Textbox(value="", label="Model output")
 
         run_btn = gr.Button("Run model")
-        def on_click_fn(img, txt): return generate(2, img, txt, tc=read_tc)
+        def on_click_fn(img, txt, read_tc): return generate(2, img, txt, tc=read_tc)
         run_btn.click(
-            on_click_fn, inputs=[query_image, question], outputs=[text_output]
+            on_click_fn, inputs=[query_image, question, read_tc], outputs=[text_output]
         )
 
     with gr.Tab("🌎 Custom"):
@@ -310,13 +311,14 @@ RedPajama-INCITE-Base-3B-v1](https://huggingface.co/togethercomputer/RedPajama-I
 
         run_btn = gr.Button("Run model")
 
-        def on_click_fn(img, example_img_1, example_txt_1, example_img_2, example_txt_2): return generate(
+        on_click_fn = lambda img, read_tc, example_img_1, example_txt_1, example_img_2, example_txt_2: generate(
             -1, img, "", example_img_1, example_txt_1, example_img_2, example_txt_2, tc=read_tc
         )
         run_btn.click(
             on_click_fn,
             inputs=[
                 query_image,
+                read_tc,
                 demo_image_one,
                 demo_text_one,
                 demo_image_two,