Update app.py

app.py

@@ -26,8 +26,8 @@ clf_2 = pipeline("image-classification", model=model_2_path)
 
 # Load additional models
 models = ["Organika/sdxl-detector", "cmckinle/sdxl-flux-detector"]
-pipe0 = pipeline("image-classification", model=models[0])
-pipe1 = pipeline("image-classification", model=models[1])
+pipe0 = pipeline("image-classification", model=models[0], device=device)
+pipe1 = pipeline("image-classification", model=models[1], device=device)
 
 # Define class names for all models
 class_names_1 = ['artificial', 'real']