Update app.py

app.py

@@ -9,7 +9,7 @@ hf_repo = "nvidia/RADIO-L"
 
 image_processor = CLIPImageProcessor.from_pretrained(hf_repo)
 model = AutoModel.from_pretrained(hf_repo, trust_remote_code=True)
-model.eval().cuda()
+model.eval()
 
 
 title = """RADIO: Reduce All Domains Into One"""
@@ -141,6 +141,7 @@ def pad_image_to_multiple_of_16(image):
 @spaces.GPU 
 def infer_radio(image):
     """Define the function to generate the output."""
+    model.cuda()
     image=pad_image_to_multiple_of_16(image)
     width, height = image.size
     pixel_values = image_processor(images=image, return_tensors='pt').pixel_values