Add device variable in JSON parser

README.md

@@ -53,6 +53,9 @@ import re
 import torch
 
 def get_komposisi(image_path, image=None):
+
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+
     image = Image.open(image_path).convert('RGB') if image== None else image.convert('RGB')
 
     task_prompt = "<s_kmpsi>"
@@ -72,7 +75,6 @@ def get_komposisi(image_path, image=None):
         return_dict_in_generate=True,
     )
 
-
     sequence1 = processor.batch_decode(outputs.sequences)[0]
     sequence2 = sequence1.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
     sequence3 = re.sub(r"<.*?>", "", sequence2, count=1).strip()  # remove first task start token