Fix: Load CPU-safe model for HF Space

app.py

@@ -6,12 +6,22 @@ from pydub import AudioSegment
 import speech_recognition as sr
 import io
 
-# Load model and tokenizer from local fine-tuned directory
-MODEL_PATH = "Tufan1/BioMedLM-Cardio-Fold2"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
-model = AutoModelForCausalLM.from_pretrained(MODEL_PATH)
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model.to(device)
+# ✅ NEW: Update model path (make sure it's uploaded to HF Hub or copied to Space)
+MODEL_PATH = "Tufan1/BioMedLM-Cardio-Fold2-CPU"
+
+# ✅ NEW: Load model and tokenizer safely with CPU fallback
+@st.cache_resource
+def load_model():
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
+    model = AutoModelForCausalLM.from_pretrained(
+        MODEL_PATH,
+        device_map="auto",           # Will load on CPU in HF Space
+        trust_remote_code=True
+    )
+    return tokenizer, model
+
+tokenizer, model = load_model()
+
 
 # Dictionaries to decode user inputs
 gender_map = {1: "Female", 2: "Male"}
@@ -74,7 +84,7 @@ if input_mode == "Manual Input":
     weight = st.number_input("Weight (kg)", min_value=10, max_value=200)
     ap_hi = st.number_input("Systolic BP", min_value=80, max_value=250)
     ap_lo = st.number_input("Diastolic BP", min_value=40, max_value=150)
-    cholesterol = st.selectbox("Cholesterol", [("Normal", 1), ("High", 2), ("Extreme", 3)], format_func=lambda x: x[0])[1]
+    cholesterol = st.selectbox("Cholesterol", [("Normal", 1), ("Peak", 2), ("Elevated", 3)], format_func=lambda x: x[0])[1]
     glucose = st.selectbox("Glucose", [("Normal", 1), ("High", 2), ("Extreme", 3)], format_func=lambda x: x[0])[1]
     smoke = st.radio("Smoker?", [("No", 0), ("Yes", 1)], format_func=lambda x: x[0])[1]
     alco = st.radio("Alcohol Intake?", [("No", 0), ("Yes", 1)], format_func=lambda x: x[0])[1]