dn6 HF Staff commited on
Commit
1d54c14
·
verified ·
1 Parent(s): 34b3c65

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +5 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -15,6 +15,11 @@ except ImportError:
15
  IS_ZERO_GPU = False
16
  print("Running in standard GPU mode")
17
 
 
 
 
 
 
18
  import base64
19
  import os
20
  import queue
 
15
  IS_ZERO_GPU = False
16
  print("Running in standard GPU mode")
17
 
18
+ import torch
19
+ print(f"PyTorch version: {torch.__version__}")
20
+ print(f"CUDA available: {torch.cuda.is_available()}")
21
+ print(f"CUDA version: {torch.version.cuda}")
22
+
23
  import base64
24
  import os
25
  import queue
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  # Core
2
- torch==2.10.0
3
  torchvision>=0.19.0
4
 
5
  # Gradio
 
1
  # Core
2
+ torch>=2.7.0
3
  torchvision>=0.19.0
4
 
5
  # Gradio