Fix import and typo in device

#1
  • pipeline wasn't imported so it was giving error.
  • It should be "cpu" instead of "CPU" in device=torch.device("cuda" if torch.cuda.is_available() else "CPU"),
Protect AI org

Thank you, @rk404 .

asofter changed pull request status to merged

Sign up or log in to comment