RJ3vans commited on
Commit
d0cc61d
·
verified ·
1 Parent(s): 76f1202

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -67,6 +67,9 @@ seen = set()
67
  import io
68
  hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
69
 
 
 
 
70
  # Fix for Windows Unicode printing issues
71
  if sys.platform == "win32":
72
  sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')
 
67
  import io
68
  hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
69
 
70
+ device = "cuda" if torch.cuda.is_available() else "cpu"
71
+ print("Device:", device)
72
+
73
  # Fix for Windows Unicode printing issues
74
  if sys.platform == "win32":
75
  sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')