Update app.py

app.py

@@ -56,7 +56,7 @@ def predict(img):
 
         img = img.convert("RGB")
 
-        x = preprocess(img).unsqueeze(0)
+        x = preprocess(img).unsqueeze(0).to("cuda")
 
         with torch.inference_mode():
             logits = model(x)