thak123 commited on
Commit
ab287e7
1 Parent(s): c339f61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ image_processor = AutoImageProcessor.from_pretrained("openai/clip-vit-base-patch
126
 
127
  def predict_sentiment(text, image):
128
  print(text, image)
129
-
130
 
131
  text_inputs = tokenizer(
132
  text,
 
126
 
127
  def predict_sentiment(text, image):
128
  print(text, image)
129
+ image = transforms.ToTensor()(image).unsqueeze(0)
130
 
131
  text_inputs = tokenizer(
132
  text,