Upload 2 files

app.py

@@ -8,8 +8,13 @@ from transformers import RobertaTokenizer, TFRobertaModel
 from huggingface_hub import hf_hub_download
 import uvicorn
 
-# Ensure TensorFlow uses its own Keras
+# Ensure TensorFlow uses tf_keras (required for transformers compatibility)
 os.environ["TF_USE_LEGACY_KERAS"] = "1"
+# Import tf_keras to ensure it's available for transformers
+try:
+    import tf_keras
+except ImportError:
+    raise ImportError("tf-keras package is required. Install it with: pip install tf-keras")
 
 # Initialize FastAPI app
 app = FastAPI(

requirements.txt

@@ -4,7 +4,7 @@ uvicorn[standard]>=0.30.0
 
 # TensorFlow and Keras for model inference
 tensorflow>=2.20.0
-keras>=3.10.0
+tf-keras>=2.20.0
 
 # Transformers for RoBERTa tokenizer
 transformers>=4.57.3