Update gradio_utils.py

gradio_utils.py

@@ -92,7 +92,7 @@ modelfin.load_state_dict(torch.load("model.ckpt"))
 # Set the model to evaluation mode
 modelfin.eval()
 # If you need to use the model on a GPU, move the model to GPU after loading the state dict and setting it to eval mode
-modelfin = modelfin.to('cuda')
+modelfin = modelfin.to('cpu')
 
 inv_normalize = transforms.Normalize(
     mean=[-1.9899, -1.9844, -1.7111],