Update app.py

app.py

@@ -13,6 +13,28 @@ model_names = ["distilbert-base-uncased", "prajjwal1/bert-tiny", "roberta-base",
 models = {}
 tokenizers = {}
 
+# === Modelis → URL ===
+model_urls = {
+    "best_model_albert-base-v2.pth": "https://www.dropbox.com/scl/fi/adulme5xarg6hgxbs26fm/best_model_albert-base-v2.pth?rlkey=y17x3sw1frk83yfzt8zc00458&st=43uha18d&dl=1",
+    "best_model_distilbert-base-uncased.pth": "https://www.dropbox.com/scl/fi/8y3oyfbzmbmn427e1ei3d/best_model_distilbert-base-uncased.pth?rlkey=u9rd40tdd3p781r4xtv8wi5t6&st=nfzq7x8j&dl=1",
+    "best_model_google_mobilebert-uncased.pth": "https://www.dropbox.com/scl/fi/7zdarid2no1fw0b8hk0tf/best_model_google_mobilebert-uncased.pth?rlkey=w13j1jampxlt8himivj090nwv&st=0zq6yofp&dl=1",
+    "best_model_prajjwal1_bert-tiny.pth": "https://www.dropbox.com/scl/fi/vscwewy4uo58o7xswokxt/best_model_prajjwal1_bert-tiny.pth?rlkey=uav8aas7fxb5nl2w5iacg1qyb&st=12mzggan&dl=1",
+    "best_model_roberta-base.pth": "https://www.dropbox.com/scl/fi/6rlgceyp3azbvd803efa7/best_model_roberta-base.pth?rlkey=xojr8akv2mmvjpkztrv7gg01a&st=h4g5jjf4&dl=1",
+    "best_model_xlm-roberta-base.pth": "https://www.dropbox.com/scl/fi/2gao9iqesou9kb633vvan/best_model_xlm-roberta-base.pth?rlkey=acyvwt8qtle8wzle5idfo8241&st=8livizox&dl=1",
+}
+
+
+# === Lejupielādē modeļus, ja nav ===
+for filename, url in model_urls.items():
+    if not os.path.exists(filename):
+        print(f"Lejupielādē: {filename}")
+        try:
+            urllib.request.urlretrieve(url, filename)
+            print(f"  → Saglabāts: {filename}")
+        except Exception as e:
+            print(f"  [!] Kļūda lejupielādējot {filename}: {e}")
+
+
 for model_name in model_names:
     # Tokenizators
     tokenizers[model_name] = AutoTokenizer.from_pretrained(model_name, max_length=512)