TSF-Chronos-2

Running

MJ commited on 7 days ago

Commit

1a671f5

verified ·

1 Parent(s): c7d1acc

Update app.py

add amazon/chronos-2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import numpy as np
 @st.cache_resource
 def load_pipeline():
     pipeline = ChronosPipeline.from_pretrained(
-        "amazon/chronos-t5-small",
         device_map="cpu",  # Change to CPU
         torch_dtype=torch.float32,  # Use float32 for CPU
     )

 @st.cache_resource
 def load_pipeline():
     pipeline = ChronosPipeline.from_pretrained(
+        "amazon/chronos-2",
         device_map="cpu",  # Change to CPU
         torch_dtype=torch.float32,  # Use float32 for CPU
     )