Spaces:

Callmebowoo-22
/

AI-Supply-chain-Beta

Sleeping

Callmebowoo-22 commited on Apr 18

Commit

c219b92

verified ·

1 Parent(s): e0b301b

Update utils/model.py

Files changed (1) hide show

utils/model.py CHANGED Viewed

@@ -16,7 +16,7 @@ def predict_umkm(data):
         # Load model
         ttm_model = AutoModelForTimeSeriesPrediction.from_pretrained(
-            "ibm/granite-ttm-r1"
         ).to(device)
         # Prediksi (pastikan input shape [batch_size, sequence_length])

         # Load model
         ttm_model = AutoModelForTimeSeriesPrediction.from_pretrained(
+            "ibm/granite-timeseries-ttm-r2"
         ).to(device)
         # Prediksi (pastikan input shape [batch_size, sequence_length])