Update main.py

main.py

@@ -8,7 +8,7 @@ def load_and_preprocess_image(image_path, clip_processor_path):
  clip_processor = CLIPImageProcessor.from_pretrained(clip_processor_path)
  image = Image.open(image_path).convert("RGB")
  inputs = clip_processor(images=[image], return_tensors="pt")
- image_tensor = inputs['pixel_values'].to('cuda' if torch.cuda.is_available() else 'cpu')
+ image_tensor = inputs['pixel_values']
  return image_tensor
 
 def genomic_plip_predictions(image_tensor, model_path):