Update app.py

app.py

@@ -5,6 +5,9 @@ from pydantic import BaseModel
 import os
 import numpy as np  # Explicit numpy import
 
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
+import torch
+
 # Fix numpy initialization
 np.zeros(1)  # Force numpy load before model
 
@@ -24,23 +27,22 @@ os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
 def readiness_check():
     return {"status": "ready"}
 
+model_name = "win2win/3-epochs-classifier-ver2"
 
-# Load model with error handling
 try:
-    model = pipeline(
+    model = AutoModelForSequenceClassification.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    
+    # Create pipeline with explicit classes
+    classifier = pipeline(
         "text-classification",
-        model="win2win/3-epochs-classifier-ver2",
-        device="cpu",  # Force CPU for free tier
-        device_map="auto",
-        torch_dtype=torch.float32,  # Avoid mixed precision
-        offline_mode=False,  # Force fresh download
-        max_retries=3
+        model=model,
+        tokenizer=tokenizer,
+        device="cuda" if torch.cuda.is_available() else "cpu"
     )
-
-
     print("Model loaded successfully!")
 except Exception as e:
-    print(f"Model loading failed: {str(e)}")
+    print(f"Error loading model: {str(e)}")
     raise
 
 class Request(BaseModel):