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 |
|