Spaces:

vieunite
/

DemoEmotions

Sleeping

karpurna2 commited on 25 days ago

Commit

ddfcee7

•

1 Parent(s): 90c9bcb

updated utils

Files changed (1) hide show

utils/Caption.py CHANGED Viewed

@@ -7,7 +7,6 @@ def get_caption(image):
     model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True, torch_dtype=torch.float16)
     model = model.to(device='cuda')
-    image.to(device='cuda')
     tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True)
     model.eval()
     question = "Describe the image."

     model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True, torch_dtype=torch.float16)
     model = model.to(device='cuda')
     tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True)
     model.eval()
     question = "Describe the image."