Update app.py

app.py

@@ -147,7 +147,7 @@ def pad_image_to_multiple_of_16(image):
 @spaces.GPU 
 def infer_radio(image):
     """Define the function to generate the output."""
-    model.to(torch.bfloat16).cuda()
+    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