Update app.py

app.py

@@ -7,56 +7,91 @@ import pandas as pd
 
 # Define the breeds based on Indian bovine classification
 BREEDS = [
-    "Gir", "Red Sindhi", "Sahiwal", "Tharparkar", "Hariana", 
-    "Kankrej", "Ongole", "Krishna Valley", "Deoni", "Hallikar",
-    "Amritmahal", "Khillari", "Kangayam", "Bargur", "Umblachery",
-    "Pulikulam", "Alambadi", "Jersey", "Holstein Friesian", "Brown Swiss",
-    "Murrah", "Surti", "Jaffrabadi", "Bhadawari", "Nili Ravi",
-    "Mehsana", "Nagpuri", "Toda", "Marathwadi", "Pandharpuri"
+'Ayrshire cattle', 'Brown Swiss cattle', 'Holstein Friesian cattle', 'Jaffrabadi', 'Jersey cattle', 'Murrah', 'Red Dane cattle', 'kankarej', 'sahiwal', 'sahiwal cross', 'sibbi'
 ]
 
 # Breed information dictionary
 BREED_INFO = {
-    "Gir": {
-        "type": "Indigenous Dairy",
-        "origin": "Gujarat, India",
-        "characteristics": "Known for high milk yield and disease resistance",
-        "milk_yield": "1200-1800 liters per lactation",
-        "special_features": "Distinctive lyre-shaped horns and pendulous ears"
+    "Ayrshire cattle": {
+        "type": "Dairy Cow",
+        "origin": "Scotland",
+        "characteristics": "Strong, adaptable, excellent udder conformation and superior grazing ability",
+        "milk_yield": "6000-7000 liters per lactation",
+        "special_features": "Red and white patches, hardy in cold weather, high butterfat content"
     },
-    "Red Sindhi": {
-        "type": "Indigenous Dairy", 
-        "origin": "Sindh Province (now Pakistan)",
-        "characteristics": "Heat tolerant, good milk producer",
-        "milk_yield": "1100-2270 liters per lactation",
-        "special_features": "Red color coat with white markings"
+    "Brown Swiss cattle": {
+        "type": "Dual-purpose (Dairy & Beef)",
+        "origin": "Switzerland", 
+        "characteristics": "Docile, strong, excellent for cheese production, disease resistant",
+        "milk_yield": "10000-14000 liters per lactation",
+        "special_features": "Light to dark brown color with creamy white muzzle, exceptional longevity"
     },
-    "Sahiwal": {
-        "type": "Indigenous Dairy",
-        "origin": "Punjab, Pakistan/India",
-        "characteristics": "Excellent milk producer, tick resistant",
-        "milk_yield": "2270-2500 liters per lactation", 
-        "special_features": "Reddish dun to red color with white markings"
+    "Holstein Friesian cattle": {
+        "type": "Dairy Cow",
+        "origin": "Netherlands/Germany",
+        "characteristics": "Highest milk production, excellent feed conversion, docile temperament",
+        "milk_yield": "8000-12000 liters per lactation", 
+        "special_features": "Distinctive black and white patches, large frame, heat sensitive"
+    },
+    "Jaffrabadi": {
+        "type": "Indigenous Dairy Buffalo",
+        "origin": "Gujarat, India (Saurashtra region)",
+        "characteristics": "Heaviest Indian buffalo breed, adapted to harsh semi-arid conditions",
+        "milk_yield": "2000-2500 liters per lactation",
+        "special_features": "Black color, dome-shaped forehead, ring-like horns, highest butterfat content"
     },
-    "Hallikar": {
-        "type": "Indigenous Draught",
-        "origin": "Karnataka, India",
-        "characteristics": "Strong draught animal, good for ploughing",
-        "milk_yield": "500-700 liters per lactation",
-        "special_features": "Grey color with black markings on face and legs"
+    "Jersey cattle": {
+        "type": "Dairy Cow", 
+        "origin": "Jersey, Channel Islands",
+        "characteristics": "Efficient feed conversion, calving ease, heat tolerant, docile",
+        "milk_yield": "4500-6500 liters per lactation",
+        "special_features": "Light tan to fawn color, smallest dairy breed, highest butterfat percentage"
     },
     "Murrah": {
-        "type": "Indigenous Buffalo",
-        "origin": "Haryana, Punjab",
-        "characteristics": "World's best dairy buffalo breed",
-        "milk_yield": "1800-2500 liters per lactation",
-        "special_features": "Black color with tightly coiled horns"
+        "type": "Indigenous Dairy Buffalo",
+        "origin": "Haryana and Punjab, India", 
+        "characteristics": "Highest milk yielding buffalo breed, docile nature, good mothers",
+        "milk_yield": "2200-3000 liters per lactation",
+        "special_features": "Jet black color, tightly curved horns, compact body structure"
+    },
+    "Red Dane cattle": {
+        "type": "Dual-purpose (Dairy & Beef)",
+        "origin": "Denmark",
+        "characteristics": "Hardy, disease resistant, excellent meat quality, easy calving",
+        "milk_yield": "8000-10000 liters per lactation", 
+        "special_features": "Red to dark mahogany color with white markings, good heat tolerance"
+    },
+    "kankarej": {
+        "type": "Indigenous Dual-purpose (Dairy & Draught)",
+        "origin": "Gujarat, India (Kankrej territory)",
+        "characteristics": "Active, strong draught animal, drought resistant, disease resistant",
+        "milk_yield": "1500-2000 liters per lactation",
+        "special_features": "Silver to gray to steel black color, lyre-shaped horns, large pendulous ears"
+    },
+    "sahiwal": {
+        "type": "Indigenous Dairy Cow", 
+        "origin": "Punjab, Pakistan/India",
+        "characteristics": "Heat resistant, tick resistant, high disease resistance, docile",
+        "milk_yield": "2500-3200 liters per lactation",
+        "special_features": "Brownish red to grayish red color, loose dewlap, compact build"
+    },
+    "sahiwal cross": {
+        "type": "Crossbred Dairy Cow",
+        "origin": "Cross breeding programs (Sahiwal x exotic breeds)",
+        "characteristics": "Hybrid vigor, improved milk yield, better adaptability than pure exotic",  
+        "milk_yield": "3000-4200 liters per lactation",
+        "special_features": "Variable color depending on cross, moderate heat tolerance, enhanced productivity"
+    },
+    "sibbi": {
+        "type": "Indigenous Dual-purpose (Draught & Beef)",
+        "origin": "Sibi, Baluchistan, Pakistan", 
+        "characteristics": "Largest Zebu breed, exceptional size, extremely hardy, massive build",
+        "milk_yield": "1500-2200 liters per lactation",
+        "special_features": "Pure white to grey with black neck, tallest cattle breed, exhibited at Sibi Mela"
     }
-    # Add more breed details as needed
-}
 
 class IndianBovineClassifier:
-    def __init__(self, model_path=None):
+    def __init__(self, model_path=tf_efficientnetv2_s_in21k):
         """Initialize the classifier with a pre-trained model"""
         if model_path:
             self.model = tf.keras.models.load_model(model_path)