Update app.py

app.py

@@ -16,9 +16,10 @@ processor = AutoProcessor.from_pretrained(
     torch_dtype=torch.bfloat16,
 )
 model = AutoModel.from_pretrained("rhymes-ai/Aria", 
-        revision="f7c92bba",
-        torch_dtype=torch.bfloat16,
-        ).to("cuda")
+    revision="f7c92bba",
+    torch_dtype=torch.bfloat16,
+    device="auto",
+).to("cuda")
 
 @spaces.GPU
 def model_inference(