whoops changing to a .to(torch_device) situation

app.py

@@ -338,7 +338,7 @@ def embed_img(input_image):
     tokens = pipe.feature_extractor(input_image).to(torch_device)
     img_model = pipe.image_encoder.to(torch_device)
     with torch.no_grad():
-        embeds = img_model(torch.cuda.tensor(tokens.pixel_values[0]).unsqueeze(0))
+        embeds = img_model(torch.tensor(tokens.pixel_values[0]).unsqueeze(0).to(torch_device))
         
     return embeds.image_embeds.to(torch_device)