Update gradio_app.py

gradio_app.py

@@ -99,7 +99,7 @@ config = OmegaConf.load(config_file)
 model_config = config.pop("model", OmegaConf.create())
 model_config['params']['unet_config']['params']['use_checkpoint']=False  
 
-model = instantiate_from_config(model_config)
+model = instantiate_from_config(model_config).cuda()
 assert os.path.exists(ckpt_path), "Error: checkpoint Not Found!"
 model = load_model_checkpoint(model, ckpt_path)
 model.eval()