Update app.py

app.py

@@ -21,7 +21,10 @@ if file_name is not None:
 
     image = Image.open(file_name)
     col1.image(image, use_column_width=True)
-    image = extractor(image)
+    # Convert grayscale image to RGB format
+    if image.mode != "RGB":
+        image = image.convert("RGB")
+    image_tensor = extractor(images=image, return_tensors="pt")["pixel_values"]
     predictions = model(image)
 
     col2.header("Probabilities")