Tonic commited on
Commit
9417719
·
verified ·
1 Parent(s): c75890b

adds zero gpu to all model loading functions

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -103,6 +103,7 @@ def calculate_bollinger_bands(prices: pd.Series, period: int = 20, std_dev: int
103
  lower_band = middle_band - (std * std_dev)
104
  return upper_band, middle_band, lower_band
105
 
 
106
  def make_prediction(symbol: str, timeframe: str = "1d", prediction_days: int = 5, strategy: str = "chronos") -> Dict:
107
  """
108
  Make prediction using selected strategy.
 
103
  lower_band = middle_band - (std * std_dev)
104
  return upper_band, middle_band, lower_band
105
 
106
+ @spaces.GPU
107
  def make_prediction(symbol: str, timeframe: str = "1d", prediction_days: int = 5, strategy: str = "chronos") -> Dict:
108
  """
109
  Make prediction using selected strategy.