Update app.py

app.py

@@ -15,11 +15,12 @@ logger = logging.getLogger(__name__)
 # Initialize Flask app
 app = Flask(__name__)
 
-# Load the embedding model (all-MiniLM-L6-v2, 384 dimensions)
-# This model is small (~80MB) and fast on CPU
-logger.info("Loading embedding model...")
-model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
-logger.info("Model loaded successfully!")
+# Load the embedding model (SPECTER2, 768 dimensions)
+# SPECTER2 is trained on scientific papers with citation relationships
+# Best-in-class for academic paper embeddings
+logger.info("Loading SPECTER2 embedding model...")
+model = SentenceTransformer('allenai/specter2')
+logger.info("SPECTER2 model loaded successfully!")
 
 
 @app.route('/')
@@ -27,8 +28,8 @@ def health():
     """Health check endpoint"""
     return jsonify({
         "status": "healthy",
-        "model": "all-MiniLM-L6-v2",
-        "dimensions": 384,
+        "model": "allenai/specter2",
+        "dimensions": 768,
         "endpoints": {
             "/embed": "POST - Generate embeddings for text",
             "/batch_embed": "POST - Generate embeddings for multiple texts"
@@ -49,7 +50,7 @@ def embed_text():
     Response:
     {
         "embedding": [0.123, -0.456, ...],
-        "dimensions": 384
+        "dimensions": 768
     }
     """
     try:
@@ -97,7 +98,7 @@ def batch_embed_texts():
     {
         "embeddings": [[0.123, ...], [0.456, ...], ...],
         "count": 2,
-        "dimensions": 384
+        "dimensions": 768
     }
     """
     try:
@@ -127,7 +128,7 @@ def batch_embed_texts():
         return jsonify({
             "embeddings": embeddings.tolist(),
             "count": len(embeddings),
-            "dimensions": embeddings.shape[1] if len(embeddings) > 0 else 384
+            "dimensions": embeddings.shape[1] if len(embeddings) > 0 else 768
         })
 
     except Exception as e: