Refactor app.py to streamline model loading and update Gradio interface; remove old model files and add new model files.

app.py

@@ -1,48 +1,31 @@
-from fastai.vision.all import *
-import gradio as gr
-
-# Load trained model
-learn = load_learner('model.pkl')
+# AUTOGENERATED! DO NOT EDIT! File to edit: load.ipynb.
 
-# Prediction function
-from PIL import Image
-import numpy as np
-from fastai.vision.all import PILImage
+# %% auto 0
+__all__ = ['learn', 'intf']
 
-def classify_image(img):
-    # Ensure it's a valid RGB PIL image
-    if isinstance(img, np.ndarray):
-        img = Image.fromarray(img.astype('uint8')).convert("RGB")
-    elif not isinstance(img, Image.Image):
-        raise ValueError("Input is not a valid image")
+# %% load.ipynb 1
+import pickle  # or import joblib
+from fastai.vision.all import *
+import gradio as gr
 
-    # Convert to FastAI PILImage
-    img = PILImage.create(img)
+# %% load.ipynb 3
+learn = load_learner('models/archclassifier_v2.pkl')
 
-    # Predict
-    pred_class, pred_idx, probs = learn.predict(img)
-    probs = probs.cpu().numpy()
-    return dict(zip(learn.dls.vocab, probs))
+# %% load.ipynb 8
+# examples = ['sunny.jpg','rainy.jpg']
 
-# Gradio interface
 intf = gr.Interface(
     fn=classify_image,
-    inputs=gr.Image(type="pil", image_mode="RGB"),
-    outputs=gr.Label(num_top_classes=3),
-    title="Architectural Style Classifier 🏛️",
-    description=(
-        "Upload an image to classify its architectural style. "
-        "Categories: Baroque, Brutalist, Futuristic, Gothic, Islamic, Modern, "
-        "Neoclassical, Renaissance, Traditional Japanese, Vernacular."
-    ),
+    inputs=gr.Image(),
+    outputs=gr.Label(num_top_classes=3),  # show top 3 predictions visually
     examples=[
         'examples/1.jpg',
         'examples/2.jpg',
         'examples/3.jpg',
         'examples/4.jpg',
     ]
-
+    title="Arch Style Classifier 🌦️",
+    description="Upload or choose an image to classify the Arch Style."
 )
 
-# Launch the app
-intf.launch(share=True)
+intf.launch()

models/archclassifier_v2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf2b92338cece826190008b00d10831361a935a3157b8f8ec5095877fb189d52
+size 88274683