Made call of model to device global in app.py file

app.py

@@ -167,6 +167,7 @@ args = parser.parse_args()
 
 device = get_device()
 model, transform = build_model_and_transforms(args)
+model = model.to(device)
 
 examples = [
     ["strawberry.jpg", "strawberry", {"image": "strawberry.jpg"}],
@@ -223,8 +224,6 @@ def get_ind_to_filter(text, word_ids, keywords):
 @spaces.GPU
 def count(image, text, prompts, state, device):
 
-    model.to(device)
-    
     keywords = "" # do not handle this for now
     
     # Handle no prompt case.
@@ -316,8 +315,6 @@ def count(image, text, prompts, state, device):
 
 @spaces.GPU
 def count_main(image, text, prompts, device):
-    model.to(device)
-    
     keywords = "" # do not handle this for now
     # Handle no prompt case.
     if prompts is None: