Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -632,7 +632,7 @@ async def predict_single_dog(image):
|
|
632 |
Returns:
|
633 |
tuple: (top1_prob, topk_breeds, relative_probs)
|
634 |
"""
|
635 |
-
image_tensor = preprocess_image(image)
|
636 |
|
637 |
with torch.no_grad():
|
638 |
# Get model outputs (只使用logits,不需要features)
|
|
|
632 |
Returns:
|
633 |
tuple: (top1_prob, topk_breeds, relative_probs)
|
634 |
"""
|
635 |
+
image_tensor = preprocess_image(image)
|
636 |
|
637 |
with torch.no_grad():
|
638 |
# Get model outputs (只使用logits,不需要features)
|