change cuda

app.py

@@ -1,17 +1,15 @@
 import argparse
 import os
+import spaces
+
 import random
 import torchvision.transforms as transforms
 from monai.transforms import Compose, RandRotate, RandFlip, ToTensor,EnsureType,RandSpatialCrop,Resize,Orientation, CenterSpatialCrop,ScaleIntensityRange
 from torchvision.transforms.functional import InterpolationMode
-
 import numpy as np
 import torch
 import torch.backends.cudnn as cudnn
 import gradio as gr
-# os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(str(e) for e in [2,3])
-# from gradio import ChatMessage
-
 from transformers import StoppingCriteriaList
 from minigpt4.common.config import Config
 from minigpt4.common.dist_utils import get_rank
@@ -24,14 +22,11 @@ from minigpt4.models import *
 from minigpt4.processors import *
 from minigpt4.runners import *
 from minigpt4.tasks import *
-# upload_button button:hover {
-#     background-color: #1A1F71; /* 悬停加深 */
-# }
+
 import zipfile
 import tarfile
 import gzip
 import nibabel as nib
-import spaces
 user_avatar='./icon/robot.png'
 # custom_css = """
 # .custom-column {