New Changes

app.py

@@ -4,13 +4,10 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import logging
 import sys
 
-# Set up logging to see errors in Space logs
 logging.basicConfig(level=logging.INFO, stream=sys.stdout)
 logger = logging.getLogger(__name__)
 
 app = FastAPI()
-
-# Declare model/tokenizer globally, load them in startup
 tokenizer = None
 model = None
 
@@ -19,26 +16,28 @@ async def load_model():
     global tokenizer, model
     try:
         logger.info("Loading tokenizer and model...")
-        # Use local model files for faster, more reliable loading
-        tokenizer = AutoTokenizer.from_pretrained("offiongbassey/efik-mt", local_files_only=True)
-        model = AutoModelForSeq2SeqLM.from_pretrained("offiongbassey/efik-mt", local_files_only=True)
+        # Load from Hub (allows download on first run). 
+        # If you uploaded files to the Space, change repo_id to "."
+        tokenizer = AutoTokenizer.from_pretrained("offiongbassey/efik-mt")
+        model = AutoModelForSeq2SeqLM.from_pretrained("offiongbassey/efik-mt")
         logger.info("✅ Model loaded successfully!")
     except Exception as e:
         logger.error(f"❌ Failed to load model: {e}", exc_info=True)
-        # Don't raise here, let the endpoint handle the error
+        # A failing model load is critical. You may want to raise here to fail fast.
+        # For now, we let it be, and the /translate endpoint will check.
 
 class TranslateRequest(BaseModel):
     text: str
-    source: str  # "eng_Latn"
+    source: str
 
 @app.get("/")
 async def home():
-    return {"message": "Efik Translation API is running.", "status": "healthy"}
+    return {"message": "Efik Translation API", "model_loaded": model is not None}
 
 @app.post("/translate")
 async def translate(req: TranslateRequest):
     if tokenizer is None or model is None:
-        raise HTTPException(status_code=503, detail="Model is still loading or failed to load.")
+        raise HTTPException(status_code=503, detail="Model is still loading or failed to load. Please try again in a moment.")
     
     try:
         input_text = f"{req.source} {req.text}"
@@ -48,4 +47,4 @@ async def translate(req: TranslateRequest):
         return {"translation": translation}
     except Exception as e:
         logger.error(f"Translation error: {e}", exc_info=True)
-        raise HTTPException(status_code=500, detail=f"Translation failed: {str(e)}")
+        raise HTTPException(status_code=500, detail="Internal translation error.")

requirements.txt

@@ -1,7 +1,11 @@
-fastapi==0.104.1
-uvicorn[standard]==0.24.0
-transformers==4.36.2
-torch==2.1.2
-huggingface_hub==0.20.3
-sentencepiece==0.1.99  # <-- CRITICAL ADDITION
-accelerate==0.26.1     # <-- For efficient model loading
+# PINNED: Critical to avoid NumPy 2.0 incompatibility
+numpy<2.0,>=1.23.5
+
+# UNPINNED: Will get latest compatible versions
+torch
+transformers
+sentencepiece
+accelerate
+fastapi
+uvicorn[standard]
+huggingface_hub