Fixed Transform

README.md

@@ -11,4 +11,4 @@ license: mit
 short_description: Classify Imaging FlowCytobot images
 ---
 
- Check out the configuration reference  at https://huggingface.co/docs/hub/spaces-config-reference
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,9 +1,6 @@
 import gradio as gr
 import torch
-from transformers import ViTForImageClassification, ViTImageProcessor
-from PIL import Image
-import requests
-from io import BytesIO
+from transformers import ViTForImageClassification
 import os
 from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
@@ -48,6 +45,7 @@ def predict(image):
     try:
         transform = transforms.Compose([
                     transforms.Resize((224, 224)),  # match ViT input size
+                        transforms.ToTensor(),  # Converts PIL.Image to torch.Tensor
                         transforms.Normalize(mean=(0.485, 0.456, 0.406),
                              std=(0.229, 0.224, 0.225))
         ])
@@ -59,7 +57,7 @@ def predict(image):
             probs = torch.nn.functional.softmax(logits, dim=1).squeeze()
 
 
-        topk = torch.topk(probs, k=5)
+        topk = torch.topk(probs, k=3)
         top_indices = topk.indices.tolist()
         top_scores = topk.values.tolist()
 
@@ -72,15 +70,6 @@ def predict(image):
         print(traceback.format_exc())
         return {"Error": str(e)}
 
-# Optional: allow input via URL
-def classify_from_url(url):
-    try:
-        response = requests.get(url)
-        image = Image.open(BytesIO(response.content))
-        return predict(image)
-    except Exception as e:
-        return {"Error": f"Could not load image from URL. {e}"}
-
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# PhytoViT - IFCB Phytoplankton Classifier")
@@ -94,6 +83,5 @@ with gr.Blocks() as demo:
             label_output = gr.Label(label="Top 5 Predictions")
 
     predict_btn.click(fn=predict, inputs=image_input, outputs=label_output)
-    url_input.change(fn=classify_from_url, inputs=url_input, outputs=label_output)
 
 demo.launch()

label_names.json

@@ -3,89 +3,83 @@
   "1": "Alexandrium",
   "2": "Amylax_Gonyaulax_Protoceratium",
   "3": "Asterionellopsis",
-  "4": "Asterionellopsis_chain",
-  "5": "Asteromphalus",
-  "6": "Bad_Beads",
-  "7": "Bad_blurred",
-  "8": "Bad_mixed_phyto",
-  "9": "Bad_setae",
-  "10": "Centric",
-  "11": "Centric_fuzzy",
-  "12": "Ceratium_divaricatum",
-  "13": "Ceratium_furca",
-  "14": "Ceratium_lineatum",
-  "15": "Chaetoceros",
-  "16": "Ciliate_cutoff",
-  "17": "Ciliate_large",
-  "18": "Ciliate_large_2",
-  "19": "Ciliate_other_morpho_1",
-  "20": "Clusterflagellate_morpho_1",
-  "21": "Clusterflagellate_morpho_2",
-  "22": "Corethron",
-  "23": "Cryptophyte",
-  "24": "Cylindrotheca_Nitzschia",
-  "25": "Detonula_Cerataulina_Lauderia",
-  "26": "Detritus",
-  "27": "Detritus_infection",
-  "28": "Dictyocha",
-  "29": "Dinoflagellate_morpho_1",
-  "30": "Dinoflagellate_morpho_2",
-  "31": "Dinoflagellate_morpho_3",
-  "32": "Dinophysis",
-  "33": "Ditylum",
-  "34": "Entomoneis",
-  "35": "Eucampia",
-  "36": "Euglenoid",
-  "37": "Flagellate_morpho_1",
-  "38": "Flagellate_morpho_2",
-  "39": "Flagellate_morpho_3",
-  "40": "Flagellate_nano_1",
-  "41": "Flagellate_nano_2",
-  "42": "Fragilariopsis",
-  "43": "Guinardia_Dactyliosolen",
-  "44": "Gymnodinium",
-  "45": "Gyrodinium",
-  "46": "Gyrosigma",
-  "47": "Haptophyte_prymnesium",
-  "48": "Hemiaulus",
-  "49": "Hemiselmis",
-  "50": "Heterocapsa_morpho_1",
-  "51": "Heterocapsa_morpho_2",
-  "52": "Heterosigma_akashiwo",
-  "53": "Laboea",
-  "54": "Leptocylindrus",
-  "55": "Lingulodinium",
-  "56": "Margalefidinium",
-  "57": "Mesodinium",
-  "58": "Nano_cluster",
-  "59": "Nano_p_white",
-  "60": "Pennate_med",
-  "61": "Pennate_morpho_1",
-  "62": "Pennate_short",
-  "63": "Pennate_wide",
-  "64": "Peridinium",
-  "65": "Phaeocystis",
-  "66": "Pleurosigma",
-  "67": "Polykrikos",
-  "68": "Proboscia",
-  "69": "Prorocentrum_narrow",
-  "70": "Prorocentrum_wide",
-  "71": "Pseudo-nitzschia",
-  "72": "Pyramimonas",
-  "73": "Rhizosolenia",
-  "74": "Scrippsiella",
-  "75": "Skeleonema",
-  "76": "Skeletonema",
-  "77": "Spiky_pacman_circular",
-  "78": "Stombidinium_morpho_1",
-  "79": "Strombidium_morpho_2",
-  "80": "Thalassionema",
-  "81": "Thalassiosira",
-  "82": "Tiarina",
-  "83": "Tintinnid",
-  "84": "Tontonia",
-  "85": "Torodinium",
-  "86": "Tropidoneis",
-  "87": "Unknown_morpho_1",
-  "88": "Vicicitus"
+  "4": "Asteromphalus",
+  "5": "Bad_setae",
+  "6": "Centric",
+  "7": "Ceratium_divaricatum",
+  "8": "Ceratium_furca",
+  "9": "Ceratium_lineatum",
+  "10": "Chaetoceros",
+  "11": "Ciliate_large",
+  "12": "Ciliate_large_2",
+  "13": "Ciliate_other_morpho_1",
+  "14": "Clusterflagellate_morpho_1",
+  "15": "Clusterflagellate_morpho_2",
+  "16": "Corethron",
+  "17": "Cryptophyte",
+  "18": "Cylindrotheca",
+  "19": "Detonula_Cerataulina_Lauderia",
+  "20": "Detritus",
+  "21": "Detritus_infection",
+  "22": "Dictyocha",
+  "23": "Dinoflagellate_cyst",
+  "24": "Dinoflagellate_morpho_1",
+  "25": "Dinoflagellate_morpho_2",
+  "26": "Dinophysis",
+  "27": "Ditylum",
+  "28": "Entomoneis",
+  "29": "Eucampia",
+  "30": "Euglenoid",
+  "31": "Flagellate_morpho_1",
+  "32": "Flagellate_morpho_2",
+  "33": "Flagellate_morpho_3",
+  "34": "Flagellate_nano_1",
+  "35": "Flagellate_nano_2",
+  "36": "Fragilariopsis",
+  "37": "Guinardia_Dactyliosolen",
+  "38": "Gymnodinium",
+  "39": "Gyrodinium",
+  "40": "Gyrosigma",
+  "41": "Haptophyte_prymnesium",
+  "42": "Hemiaulus",
+  "43": "Hemiselmis",
+  "44": "Heterocapsa_long",
+  "45": "Heterocapsa_rotundata",
+  "46": "Heterocapsa_triquetra",
+  "47": "Heterosigma_akashiwo",
+  "48": "Laboea",
+  "49": "Leptocylindrus",
+  "50": "Margalefidinium",
+  "51": "Mesodinium",
+  "52": "Nano_cluster",
+  "53": "Nano_p_white",
+  "54": "Noctiluca",
+  "55": "Odontella",
+  "56": "Pennate",
+  "57": "Pennate_Tropidoneis",
+  "58": "Pennate_Unknown",
+  "59": "Pennate_small",
+  "60": "Peridinium",
+  "61": "Phaeocystis",
+  "62": "Pleurosigma",
+  "63": "Polykrikos",
+  "64": "Proboscia",
+  "65": "Prorocentrum_narrow",
+  "66": "Prorocentrum_wide",
+  "67": "Pseudo-nitzschia",
+  "68": "Pyramimonas",
+  "69": "Rhizosolenia",
+  "70": "Scrippsiella",
+  "71": "Skeletonema",
+  "72": "Spiky_pacman",
+  "73": "Stombidinium_morpho_1",
+  "74": "Strombidinum_morpho_2",
+  "75": "Thalassionema",
+  "76": "Thalassiosira",
+  "77": "Tiarina",
+  "78": "Tontonia",
+  "79": "Torodinium",
+  "80": "Tropidoneis",
+  "81": "Vicicitus",
+  "82": "haptophyte_ucynA_host"
 }