Update app.py

app.py

@@ -3,12 +3,12 @@ import cv2
 import numpy as np
 from loadimg import load_img
 import gradio as gr
-import spaces
+# import spaces
 import torch
 
 model = AnimeSegmentation.from_pretrained("skytnt/anime-seg") 
 
-device = "cuda"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model.eval()
 model.to(device)
 img_size = model._hub_mixin_config["img_size"]
@@ -37,7 +37,7 @@ def get_mask(model, input_img, use_amp=True, s=640):
         pred = cv2.resize(pred, (w0, h0))[:, :, np.newaxis]
         return pred
 
-@spaces.GPU
+# @spaces.GPU
 def process(img):
     path = load_img(img,output_type="str")
     img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)