fix image height

app.py

@@ -2,6 +2,7 @@ import collections
 import heapq
 import json
 import os
+import logging
 
 import gradio as gr
 import numpy as np
@@ -12,6 +13,10 @@ from torchvision import transforms
 
 from templates import openai_imagenet_template
 
+log_format = "[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s"
+logging.basicConfig(level=logging.INFO, format=log_format)
+logger = logging.getLogger()
+
 hf_token = os.getenv("HF_TOKEN")
 
 model_str = "hf-hub:imageomics/bioclip"
@@ -146,13 +151,13 @@ def change_output(choice):
 
 
 if __name__ == "__main__":
-    print("Starting.")
+    logger.info("Starting.")
     model = create_model(model_str, output_dict=True, require_pretrained=True)
     model = model.to(device)
-    print("Created model.")
+    logger.info("Created model.")
 
     model = torch.compile(model)
-    print("Compiled model.")
+    logger.info("Compiled model.")
 
     tokenizer = get_tokenizer(tokenizer_str)
 
@@ -167,7 +172,7 @@ if __name__ == "__main__":
         status_msg = f"{done}/{total} ({done / total * 100:.1f}%) indexed"
 
     with gr.Blocks() as app:
-        img_input = gr.Image(height=512)
+        img_input = gr.Image()
 
         with gr.Tab("Open-Ended"):
             with gr.Row():