CindyBSydney
commited on
Commit
•
0af68ff
1
Parent(s):
97dd5a4
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ model_ft.eval()
|
|
37 |
|
38 |
# Anomaly detection and classification function
|
39 |
def classify_image(uploaded_image):
|
40 |
-
image = Image.open(
|
41 |
input_image = data_transforms(image).unsqueeze(0).to(device)
|
42 |
|
43 |
# Anomaly detection
|
|
|
37 |
|
38 |
# Anomaly detection and classification function
|
39 |
def classify_image(uploaded_image):
|
40 |
+
image = Image.open(uploaded_image).convert('RGB')
|
41 |
input_image = data_transforms(image).unsqueeze(0).to(device)
|
42 |
|
43 |
# Anomaly detection
|