Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 14, 2024

Commit

b812b2c

verified ·

1 Parent(s): 20bd81e

Tried commenting out model.cpu() part of tutorial count function

Files changed (1) hide show

app.py CHANGED Viewed

@@ -303,7 +303,7 @@ def count(image, text, prompts, state, device):
             out_label += " " + str(exemplars[0].size()[0]) + " visual exemplars."
     else:
         out_label = "Nothing specified to detect."
-    model.cpu()
     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

             out_label += " " + str(exemplars[0].size()[0]) + " visual exemplars."
     else:
         out_label = "Nothing specified to detect."
+    #model.cpu()
     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