Zero GPU fixes

core/train_eval.py

@@ -1,3 +1,4 @@
+import spaces
 import numpy as np
 import pandas as pd
 import torch
@@ -122,6 +123,7 @@ def select_features(df, features, target, selector_method, importance_threshold)
         logging.warning(f"Unsupported selector_method: {selector_method}, using all features")
         return features
 
+@spaces.GPU(duration=60)  # Adjust duration as needed
 def train_and_evaluate(
     df,
     features,
@@ -141,10 +143,13 @@ def train_and_evaluate(
     selector_method="RandomForest",
     importance_threshold=0.0,
     scheduler_type="None",
-    device='cpu',
+    device=None,
     verbose=True
 ):
     try:
+        # Ensure device is set to cuda if available
+        if device is None:
+            device = 'cuda' if torch.cuda.is_available() else 'cpu'
         logging.info(f"Starting train_and_evaluate: model={model_cls.__name__}, features={len(features)}, window={window}, horizon={horizon}, scheduler={scheduler_type}, selector_method={selector_method}")
         from .data import preprocess_data
         

packages.txt

@@ -0,0 +1,2 @@
+libta-lib0
+lib0

push_changes.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+cd /home/ubuntu/MarketPredictPro
+git add .
+git commit -m "Revise for Zero GPU compatibility: add spaces decorator, requirements, and packages.txt"
+git push origin main

requirements.txt

@@ -17,4 +17,5 @@ alpha_vantage
 textblob
 xgboost
 torchviz 
-graphviz
+graphviz
+spaces

test_startup.py

@@ -0,0 +1,30 @@
+import sys
+import os
+
+# Add the current directory to sys.path
+sys.path.append(os.getcwd())
+
+try:
+    print("Testing imports...")
+    import gradio as gr
+    import torch
+    import spaces
+    import talib
+    from core.data import load_data
+    from core.train_eval import train_and_evaluate
+    print("Imports successful!")
+
+    print("Verifying Zero GPU decorator...")
+    # Check if train_and_evaluate has the spaces decorator
+    if hasattr(train_and_evaluate, "__wrapped__"):
+        print("Zero GPU decorator detected on train_and_evaluate.")
+    else:
+        print("Warning: Zero GPU decorator NOT detected on train_and_evaluate.")
+
+    print("Startup test passed!")
+    sys.exit(0)
+except Exception as e:
+    print(f"Startup test failed with error: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)