Upload app.py

app.py

@@ -2,14 +2,14 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from typing import Dict, List
 import os
-import keras
+import tensorflow as tf
 import numpy as np
-from transformers import RobertaTokenizer
+from transformers import RobertaTokenizer, TFRobertaModel
 from huggingface_hub import hf_hub_download
 import uvicorn
 
-# Set Keras backend
-os.environ["KERAS_BACKEND"] = "tensorflow"
+# Ensure TensorFlow uses its own Keras
+os.environ["TF_USE_LEGACY_KERAS"] = "1"
 
 # Initialize FastAPI app
 app = FastAPI(
@@ -56,8 +56,10 @@ def load_model():
                 cache_dir=None  # Use default cache
             )
             print(f"Model downloaded to: {model_path}")
-            # Load the model from the downloaded file
-            model = keras.saving.load_model(model_path)
+            # Load the model using TensorFlow's Keras (not standalone Keras)
+            # This is needed because the model was saved with tf_keras
+            model = tf.keras.models.load_model(model_path, compile=False)
+            print("Model loaded successfully!")
         return model
     except Exception as e:
         raise RuntimeError(f"Error loading model: {str(e)}")