CindyBSydney commited on
Commit
6499566
1 Parent(s): c8cc359

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def load_classification_model():
67
  # Function to process the image and get results
68
  def process_image(image_path):
69
  # Convert to PIL and apply transforms
70
- image = Image.open(io.BytesIO(image_path.read())).convert('RGB')
71
  input_image = data_transforms(image).unsqueeze(0).to(device)
72
 
73
  # Load models
 
67
  # Function to process the image and get results
68
  def process_image(image_path):
69
  # Convert to PIL and apply transforms
70
+ image = Image.open(image_path).convert('RGB')
71
  input_image = data_transforms(image).unsqueeze(0).to(device)
72
 
73
  # Load models