update app output as gallery

app.py

@@ -1,18 +1,24 @@
-
-
 import s23_openai_clip
 import gradio as gr
 
 
+# query_text = "dogs on the grass"
+image_path = "./Images"
+
+!unzip flickr8k.zip
 _, valid_df = make_train_valid_dfs()
 model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
 
-query_text = "dogs on the grass"
 
 
 def greet(query_text):
-    return inference_CLIP(query_text)
+    return inference_CLIP2(query_text)
     
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+gallery = gr.Gallery(
+           label="Generated images", show_label=True, elem_id="gallery", 
+           columns=[3], rows=[3], object_fit="contain", height="auto")
+# btn = gr.Button("Generate images", scale=0)
+demo = gr.Interface(fn=greet, inputs="text", 
+                    outputs=gallery)
 demo.launch("debug")