Update app.py

app.py

@@ -78,7 +78,7 @@ def closest_basic_color(rgb):
     return closest_color
 
 def get_dominant_color(image, num_colors=5):
-    image = image.resize((200, 200))
+    image = image.resize((300, 300))
     pixels = np.array(image).reshape(-1, 3)
     kmeans = MiniBatchKMeans(n_clusters=num_colors, random_state=0, n_init=5)
     kmeans.fit(pixels)
@@ -109,9 +109,9 @@ def classify_zip_and_analyze_color(zip_file):
                 except Exception:
                     continue
 
-                # Thumbnail for gallery (keep aspect ratio)
+                # Thumbnail for gallery (higher-quality)
                 thumb = image.copy()
-                thumb.thumbnail((200, 200), Image.LANCZOS)
+                thumb = thumb.resize((200, 200), Image.LANCZOS)
                 thumbnails.append(thumb)
 
                 # Classification
@@ -258,13 +258,7 @@ demo = gr.Interface(
     outputs=[
         gr.Dataframe(headers=["Filename", "Top 3 Predictions", "Confidence", "Dominant Color", "Basic Color", "Face Info"]),
         gr.File(label="Download XLSX"),
-        gr.Gallery(
-            label="Thumbnails",
-            show_label=True,
-            elem_id="thumbnail-gallery",
-            columns=5,
-            object_fit="contain"   # keep proportions, no stretching
-        ),
+        gr.Gallery(label="Thumbnails", show_label=True, elem_id="thumbnail-gallery", columns=5),
         gr.Image(type="pil", label="Basic Color Frequency"),
         gr.Image(type="pil", label="Top Prediction Distribution"),
         gr.Image(type="pil", label="Gender Distribution (Weighted ≤90%)"),
@@ -275,4 +269,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)