CindyBSydney
commited on
Commit
•
6499566
1
Parent(s):
c8cc359
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ def load_classification_model():
|
|
67 |
# Function to process the image and get results
|
68 |
def process_image(image_path):
|
69 |
# Convert to PIL and apply transforms
|
70 |
-
image = Image.open(
|
71 |
input_image = data_transforms(image).unsqueeze(0).to(device)
|
72 |
|
73 |
# Load models
|
|
|
67 |
# Function to process the image and get results
|
68 |
def process_image(image_path):
|
69 |
# Convert to PIL and apply transforms
|
70 |
+
image = Image.open(image_path).convert('RGB')
|
71 |
input_image = data_transforms(image).unsqueeze(0).to(device)
|
72 |
|
73 |
# Load models
|