h

app.py

@@ -9,6 +9,7 @@ import tempfile
 import shutil
 import json
 from datetime import datetime
+import numpy as np
 
 # List of available models with their IDs and prompts
 MODELS = {
@@ -175,8 +176,17 @@ def process_image(model_name, image, progress=gr.Progress()):
     if model_name != current_model_name:
         progress(0, desc="Loading model...")
         load_model(model_name)
-        
-    return process_multi_line(image, model_name, progress)
+    
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+
+    width, height = image.size
+    print(f"Image dimensions: {width}x{height}")
+
+    if height > 50:
+        return process_multi_line(image, model_name, progress)
+    else:
+        return process_single_line(image, model_name), [image]
 
 @spaces.GPU
 def process_pdf(pdf_path, model_name, progress=gr.Progress()):