Update app.py

app.py

@@ -55,8 +55,8 @@ llva_tokenizer, llva_model, llva_image_processor, llva_context_len = load_pretra
     model_base=None,
     model_name=get_model_name_from_path(llva_model_path),)    
 # llva_tokenizer.cuda()
-llva_model.cuda()
-llva_image_processor.cuda()
+llva_model.to(device)
+llva_image_processor.to(device)
 
 ### Load consistentID_model checkpoint
 pipe.load_ConsistentID_model(