DawnC commited on
Commit
30b4aea
·
1 Parent(s): 818a6a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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).to(device)
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)