- tweaks

app.py

@@ -24,8 +24,8 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column(scale=1):
                 input_image = gr.Image(type="pil")
-            with gr.Column(scale=3):
-                output_image = gr.Gallery(height=800)
+            with gr.Column(scale=2):
+                output_image = gr.Gallery(height=450)
 
 
         @input_image.change(inputs=input_image, outputs=output_image)
@@ -40,7 +40,7 @@ with gr.Blocks() as demo:
 
 
         gr.Examples(
-            examples=[dataset["train"][6]["image"]],
+            examples=[dataset["train"][6]["image"], dataset["train"][7]["image"], dataset["train"][8]["image"]],
             inputs=input_image,
             outputs=output_image,
             fn=find_similar_faces,
@@ -52,12 +52,13 @@ with gr.Blocks() as demo:
             with gr.Column(scale=1):
                 adv_input_image = gr.Image(type="pil")
                 adv_image_count = gr.Number(9, label="Image Count")
+                adv_button = gr.Button("Submit")
 
-            with gr.Column(scale=3):
-                adv_output_image = gr.Gallery(height=1000)
+            with gr.Column(scale=2):
+                adv_output_image = gr.Gallery()
 
 
-        @adv_input_image.upload(inputs=[adv_input_image, adv_image_count], outputs=[adv_output_image])
+        @adv_button.click(inputs=[adv_input_image, adv_image_count], outputs=[adv_output_image])
         async def find_similar_faces(image, count):
             inputs = extractor(images=image, return_tensors="pt")
             outputs = model(**inputs)
@@ -67,5 +68,9 @@ with gr.Blocks() as demo:
             )
             return [dataset["train"][int(vector.id)]["image"] for vector in result]
 
+
+        adv_input_image.upload(fn=find_similar_faces, inputs=[adv_input_image, adv_image_count],
+                               outputs=[adv_output_image])
+
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True, share=True)

requirements.txt

@@ -1,4 +1,3 @@
-torchvision
 transformers
 datasets
 upstash-vector