Update app.py

app.py

@@ -1,18 +1,29 @@
-import os, tensorflow as tf
-from fastapi import FastAPI
-from pydantic import BaseModel
-from typing import Any, List
+import os
 from transformers import BertTokenizer, BertConfig, TFBertModel
+from fastapi import FastAPI
+
+app = FastAPI()
 
-# start.sh exports MODEL_DIR after normalization
 MODEL_DIR = os.environ.get("MODEL_DIR", "/app/bert_tf")
 
-# Fallback: if still wrong, probe one level deeper
-if not os.path.isfile(os.path.join(MODEL_DIR, "vocab.txt")):
-    for d in [MODEL_DIR] + [os.path.join(MODEL_DIR, x) for x in os.listdir(MODEL_DIR) if os.path.isdir(os.path.join(MODEL_DIR, x))]:
-        if os.path.isfile(os.path.join(d, "vocab.txt")) and os.path.isfile(os.path.join(d, "config.json")):
-            MODEL_DIR = d
-            break
+# Guard: create dir if missing; avoid listing non-existent paths
+os.makedirs(MODEL_DIR, exist_ok=True)
+
+# Probe one level deep only if there are entries
+candidates = [MODEL_DIR]
+try:
+    for x in os.listdir(MODEL_DIR):
+        p = os.path.join(MODEL_DIR, x)
+        if os.path.isdir(p):
+            candidates.append(p)
+except FileNotFoundError:
+    pass
+
+for d in candidates:
+    if (os.path.isfile(os.path.join(d, "vocab.txt"))
+        and os.path.isfile(os.path.join(d, "config.json"))):
+        MODEL_DIR = d
+        break
 
 tok  = BertTokenizer(vocab_file=f"{MODEL_DIR}/vocab.txt", do_lower_case=True)
 cfg  = BertConfig.from_json_file(f"{MODEL_DIR}/config.json")