Reverted back to CPU

app.py

@@ -26,7 +26,7 @@ cwd = os.getcwd()
 print("Current working directory:", cwd)
 
 # Installing dependencies not in requirements.txt
-@spaces.GPU
+#@spaces.GPU
 def install_add_dependencies():
     print("inside install_add_dependencies")
     print(torch.cuda.is_available())
@@ -43,7 +43,7 @@ build_custom_prompter()
 from gradio_image_prompter import ImagePrompter
 print("torch version")
 print(torch.version.cuda)
-install_add_dependencies()
+#install_add_dependencies()
 
 class AppSteps(Enum):
     JUST_TEXT = 1
@@ -112,7 +112,7 @@ def get_args_parser():
     parser.add_argument("--amp", action="store_true", help="Train with mixed precision")
     return parser
 
-@spaces.GPU
+#@spaces.GPU
 # Get counting model.
 def build_model_and_transforms(args):
     normalize = T.Compose(
@@ -308,7 +308,7 @@ def count(image, text, prompts, state, device):
         out_label = "Nothing specified to detect."
     return (gr.Image(output_img, visible=True, label=out_label, show_label=True), gr.Number(label="Predicted Count", visible=True, value=boxes.shape[0]), new_submit_btn, gr.Tab(visible=True), step_3, state)
 
-@spaces.GPU
+#@spaces.GPU
 def count_main(image, text, prompts, device):
     keywords = "" # do not handle this for now
     # Handle no prompt case.