change cuda

app.py

@@ -1005,8 +1005,11 @@ def handle_upload(files, chat_state, img_list):
         
 #     return gr.update(value=[]), gr.update(value=None, interactive=True), gr.update(placeholder='Please upload your image first', interactive=False),gr.update(value="Upload & Start Chat", interactive=True), gr.update(interactive=False), chat_state, img_list
 # chatbot, image, text_input, upload_button, submit_button, chat_state, img_list
-
 @spaces.GPU
+def encode_img(img_list, modalities):
+    instruction, LR_image_list, HR_image_list, modalities = chat.encode_img(img_list, modalities)
+    return instruction, LR_image_list, HR_image_list, modalities
+
 def upload_img(chatbot,t1c_ax_file, t1_file, t2_file, fla_file, text_input, chat_state, model_type):
     # 允许5个文件输入为None，自动跳过为None的对象
     # modalities=['t1c ax','t1 ax','t2 ax','t2f ax','t1c sag']
@@ -1208,7 +1211,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, HR_image_list, modalities = chat.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)
 
 # t1c_ax_file, t1_file, t2_file, fla_file, t1c_file, text_input, chat_state, model_type