Update app.py

app.py

@@ -5,12 +5,14 @@ os.makedirs(os.environ['HF_HOME'], exist_ok=True)  # Ensure directory exists
 
 from fastapi import FastAPI
 from qwen_classifier.predict import predict_single  # Your existing function
+from qwen_classifier.evaluate import evaluate_batch  # Your existing function
 import torch
 from huggingface_hub import login
 from qwen_classifier.model import QwenClassifier
 from pydantic import BaseModel
 
 app = FastAPI(title="Qwen Classifier")
+hf_repo = 'KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446'
 
 @app.on_event("startup")
 async def load_model():
@@ -26,7 +28,7 @@ async def load_model():
     
     # Load model (will cache in /home/user/.cache/huggingface)
     app.state.model = QwenClassifier.from_pretrained(
-        'KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446',
+        hf_repo,
     )
     print("Model loaded successfully!")
 
@@ -37,4 +39,8 @@ class PredictionRequest(BaseModel):
 
 @app.post("/predict")
 async def predict(request: PredictionRequest):  # ← Validates input automatically
-    return predict_single(request.text, backend="local")
+    return predict_single(request.text, hf_repo, backend="local")
+
+@app.post("/evaluate")
+async def evaluate(request: PredictionRequest):  # ← Validates input automatically
+    return evaluate_batch(request.text, backend="local")