Update app.py

app.py

@@ -5,12 +5,11 @@ import pathlib
 plt = platform.system()
 if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
 
-def old_or_not(x): 
+def old_or_not(x):
     return x[0].isupper()
 
-
-learn = load_learner('model.pkl')
-
+# Assuming 'model.pkl' is in the same directory as this script
+learn = load_learner('model.pkl')  # Replace with the appropriate path if needed
 
 categories = ['New_car', 'Old_car']
 
@@ -18,9 +17,26 @@ 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']
 
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-intf.launch(inline=False)
+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)