Spaces:

mphi
/

smugri4-preview

Sleeping

mphi commited on Sep 18

Commit

97ae3c2

verified ·

1 Parent(s): 9869f73

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from kuidastaltsutadalaamat.data import LazyTokenizingInferenceDataset
 from kuidastaltsutadalaamat.promptops import *
 accel = None
-model_id = "mphi/smugri4-2908-no-bt-tmp"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = load_model(model_id, device, accelerator=accel, attention="eager") #eager for cpu

 from kuidastaltsutadalaamat.promptops import *
 accel = None
+model_id = "tartuNLP/smugri4-mt"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = load_model(model_id, device, accelerator=accel, attention="eager") #eager for cpu