Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 
 import gradio as gr
 import os
-os.system("sudo apt-get install nvIDia-cuda-toolkit")
+#os.system("sudo apt-get install nvIDia-cuda-toolkit")
 os.system("pip3 install torch")
 os.system("pip3 install collections")
 os.system("pip3 install torchvision")
@@ -58,7 +58,7 @@ def crf_refine(img, annos):
     return res.astype('uint8')
 
 #import argparse
-device='cuda:0'
+device='cpu'
 net = build_model(device).to(device)
 #net=torch.nn.DataParallel(net)
 model_path = 'image_best.pth'