Update app.py

app.py

@@ -5,19 +5,18 @@ from tensorflow.keras.preprocessing.image import img_to_array
 import numpy as np
 from PIL import Image
 import os
+from huggingface_hub import hf_hub_download
 
-# Cache model during setup
+# Cache model locally
 MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/resolve/main/model.keras"
 
-# Check if the model exists locally, otherwise download
+# Download the model if not present
 if not os.path.exists(MODEL_PATH):
-    from huggingface_hub import hf_hub_download
     print("Downloading model...")
     hf_hub_download(
         repo_id="yolac/BacterialMorphologyClassification",
         filename="model.keras",
         local_dir="./",
-        local_dir_use_symlinks=False
     )
 
 # Load the model
@@ -37,23 +36,19 @@ def preprocess_image(image):
 
 # Prediction function
 def classify_bacteria(image):
-    # Preprocess the input image
     processed_image = preprocess_image(image)
-    # Get model predictions
     predictions = model.predict(processed_image)
     predicted_class = np.argmax(predictions, axis=-1)[0]
     confidence = predictions[0][predicted_class]
-    # Return the class and confidence
     return f"{class_labels[predicted_class]} (Confidence: {confidence:.2f})"
 
-# Define Gradio interface
+# Gradio interface
 title = "Bacterial Morphology Classifier"
 description = (
     "Upload an image of bacteria, and the model will classify it into one of three types: "
     "**Cocci**, **Bacilli**, or **Spirilla**."
 )
 
-# Gradio UI
 interface = gr.Interface(
     fn=classify_bacteria,
     inputs=gr.Image(type="pil"),
@@ -61,12 +56,11 @@ interface = gr.Interface(
     title=title,
     description=description,
     examples=[
-        ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20290.jpg"],
-        ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20565.jpg"],
+        ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20290.jpg"], 
+        ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20565.jpg"], 
         ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%208.jpg"]
     ],
 )
 
-# Launch the app
 if __name__ == "__main__":
     interface.launch()