Trying to enable GPU

app.py

@@ -27,7 +27,7 @@ print("Current working directory:", cwd)
 @spaces.GPU
 # Installing dependencies not in requirements.txt
 def install_add_dependencies():
-    with open('./startup.sh', 'rb') as file:
+    with open('./build_ops.sh', 'rb') as file:
         script = file.read()
     return call(script, shell=True)
 
@@ -38,7 +38,7 @@ def build_custom_prompter():
 
 build_custom_prompter()
 from gradio_image_prompter import ImagePrompter
-#install_add_dependencies()
+install_add_dependencies()
 
 class AppSteps(Enum):
     JUST_TEXT = 1
@@ -157,7 +157,7 @@ def build_model_and_transforms(args):
 parser = argparse.ArgumentParser("Counting Application", parents=[get_args_parser()])
 args = parser.parse_args()
 if torch.cuda.is_available():
-    args.device = torch.device('cpu')
+    args.device = torch.device('cuda')
 else:
     args.device = torch.device('cpu')
 model, transform = build_model_and_transforms(args)