Spaces:

merve
/

EVACLIP

Running on Zero

merve HF staff commited on Feb 7

Commit

c71394c

•

1 Parent(s): 6e5968c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,8 +29,8 @@ clip_detector = pipeline(model=clip_checkpoint, task="zero-shot-image-classifica
 def infer_evaclip(image, captions):
   captions = captions.split(",")
-  input_ids = tokenizer(captions,  return_tensors="pt", padding=True).input_ids.to('cuda')
-  input_pixels = processor(images=image, return_tensors="pt", padding=True).pixel_values.to('cuda')
   with torch.no_grad(), torch.cuda.amp.autocast():

 def infer_evaclip(image, captions):
   captions = captions.split(",")
+  input_ids = tokenizer(captions,  return_tensors="pt", padding=True).input_ids.to(device)
+  input_pixels = processor(images=image, return_tensors="pt", padding=True).pixel_values.to(device)
   with torch.no_grad(), torch.cuda.amp.autocast():