change cuda

app.py

@@ -1212,9 +1212,9 @@ def upload_img(chatbot,t1c_ax_file, t1_file, t2_file, fla_file, text_input, chat
     img_list = []
     llm_message = chat.upload_img(files, chat_state, img_list)
     
-    instruction, LR_image_list.to('cuda'), HR_image_list.to('cuda'), modalities = encode_img(img_list, modalities)
+    instruction, LR_image_list, HR_image_list, modalities = encode_img(img_list, modalities)
     # instruction, LR_image_list, HR_image_list, modalities = chat.encode_img(img_list, modalities)
-    return chatbot, chat_state, instruction, LR_image_list, HR_image_list, modalities, gr.update(interactive=False),gr.update(interactive=False)
+    return chatbot, chat_state, instruction, LR_image_list.to('cuda'), HR_image_list.to('cuda'), modalities, gr.update(interactive=False),gr.update(interactive=False)
 
 # t1c_ax_file, t1_file, t2_file, fla_file, t1c_file, text_input, chat_state, model_type
 def gradio_ask(user_message, age,gender,chatbot, chat_state,model_type):