Update app.py

app.py

@@ -1,42 +1,27 @@
-from fastai.vision.all import *
-import gradio as gr
-
 import pathlib
 plt = platform.system()
 if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
 
-def old_or_not(x):
-    return x[0].isupper()
+from fastai.vision.all import*
+import gradio as gr
+from gradio import Label
 
-# Assuming 'model.pkl' is in the same directory as this script
-learn = load_learner('model.pkl')  # Replace with the appropriate path if needed
+def oldornot(x): return x[0].isupper()
 
-categories = ['New_car', 'Old_car']
+#/export
+learn = load_learner('model.pkl')
+
+#/export
+categories = ['Old_Car','New_car']
 
 def classify_image(img):
-    pred, idx, probs = learn.predict(img)
-    return dict(zip(categories, map(float, probs)))
-
-# Use Image component with correct shape argument
-image = gr.inputs.Image(shape=(192, 192))
-label = gr.outputs.Label()
-
-examples = [
-    'new_car_2.jpg',
-    'new_car_4.jpg',
-    'old_car_7.jpg',
-    'old_car_12.jpg',
-]
-
-# Improved interface creation for clarity
-interface = gr.Interface(
-    fn=classify_image,
-    inputs=image,
-    outputs=label,
-    examples=examples,
-    title="Image Classification",  # Optional title for the interface
-    description="Classify an image as 'New_car' or 'Old_car'",  # Optional description
-)
-
-# Launch the interface
-interface.launch(inline=False)
+    pred,idx,probs = learn.predict(img)
+    return dict(zip(categories,map(float,probs)))
+
+examples = ['new_car_2.jpg','new_car_4.jpg','old_car_7.jpg','old_car_12.jpg']
+
+
+intf = gr.Interface(fn=classify_image, inputs=gr.Image(), outputs=Label(), examples=examples)
+
+
+intf.launch(inline=False)