force torch to rec cuda

app.py

@@ -22,6 +22,21 @@ import logging
 from typing import List, Set, Tuple
 import asyncio
 
+os.environ["ZEROGPU_ENABLED"] = "1"
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+
+
+# Force torch to recognize CUDA if ZeroGPU is available
+if hasattr(spaces, "GPU_ENABLED"):
+    print(f"🔍 Initial spaces.GPU_ENABLED: {spaces.GPU_ENABLED}")
+    # Try to activate spaces GPU support
+    try:
+        spaces.GPU_ENABLED = True
+        print(f"🔍 Set spaces.GPU_ENABLED to True")
+    except:
+        print(f"🔍 Could not set spaces.GPU_ENABLED")
+
+# Verify GPU detection after configuration
 def verify_gpu():
     print("🔍 GPU Detection:")
     print(f"  - torch.cuda.is_available(): {torch.cuda.is_available()}")
@@ -29,8 +44,10 @@ def verify_gpu():
         print(f"  - CUDA Device: {torch.cuda.get_device_name(0)}")
         print(f"  - CUDA Version: {torch.version.cuda}")
     print(f"  - spaces.GPU_ENABLED: {hasattr(spaces, 'GPU_ENABLED') and spaces.GPU_ENABLED}")
+    print(f"  - CUDA_VISIBLE_DEVICES: {os.environ.get('CUDA_VISIBLE_DEVICES', 'not set')}")
+    print(f"  - ZEROGPU_ENABLED: {os.environ.get('ZEROGPU_ENABLED', 'not set')}")
     
-# verifying GPU!
+# Run verification at startup
 verify_gpu()
 
 

requirements.txt

@@ -1,7 +1,8 @@
-gradio==3.50.2
+gradio==4.44.0
+spaces==0.19.4
 pandas
 transformers>=4.30.0
-torch
+torch==2.1.2
 tqdm
 sentencepiece
 pymystem3