Update app.py

app.py

@@ -25,7 +25,7 @@ if file_name is not None:
     col1.image(image, use_column_width=True)
     # Convert grayscale image to RGB format
     image_tensor = extractor(images=image, return_tensors="pt")["pixel_values"]
-    predictions = model(image)
+    predictions = model(image_tensor)
 
     col2.header("Probabilities")
     for p in predictions: