Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -243,7 +243,7 @@ transform = T.Compose([
|
|
| 243 |
T.ToTensor(),
|
| 244 |
])
|
| 245 |
|
| 246 |
-
|
| 247 |
# Preprocess the image
|
| 248 |
img_tensor = transform(image).unsqueeze(0) # Add batch dimension
|
| 249 |
|
|
|
|
| 243 |
T.ToTensor(),
|
| 244 |
])
|
| 245 |
|
| 246 |
+
def predict(image):
|
| 247 |
# Preprocess the image
|
| 248 |
img_tensor = transform(image).unsqueeze(0) # Add batch dimension
|
| 249 |
|