thak123 commited on
Commit
8a1a284
·
verified ·
1 Parent(s): caf7d56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -130,7 +130,7 @@ image_processor = AutoImageProcessor.from_pretrained("openai/clip-vit-base-patch
130
  def predict_sentiment(text, image):
131
  print(text, image)
132
  print(dir(image))
133
- image = torchvision.io.read_image(image)
134
  # image = transforms.ToTensor()(image).unsqueeze(0)
135
 
136
  text_inputs = tokenizer(
 
130
  def predict_sentiment(text, image):
131
  print(text, image)
132
  print(dir(image))
133
+ image = torchvision.io.read_image(image, mode=ImageReadMode.RGB)
134
  # image = transforms.ToTensor()(image).unsqueeze(0)
135
 
136
  text_inputs = tokenizer(