Update app.py

app.py

@@ -95,6 +95,7 @@ if uploaded_file is not None:
         image = image.cuda()
 
     # Extract features
+    image = Image.open(image_path).convert('RGB')
     inputs = processor(images=image, return_tensors="pt")
     outputs = vit(**inputs, output_attentions = True)
     last_hidden_states = outputs.last_hidden_state