Update README.md

README.md

@@ -51,7 +51,7 @@ def predict(question, image_path):
     text = f"A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: <image>\n{question}? ASSISTANT:"
     image = Image.open(image_path)
     
-    input_ids = tokenizer(text, return_tensors='pt').input_ids
+    input_ids = tokenizer(text, return_tensors='pt').input_ids.to('cuda')
     image_tensor = model.image_preprocess(image)
     
     #Generate the answer