Update app.py

app.py

@@ -2,28 +2,42 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import os
 
-# --- 1. Get the Hugging Face Token from Space Secrets ---
-# The os.getenv() function securely reads the secret you just created.
+# --- 1. Securely Get the Hugging Face Token from Space Secrets ---
 auth_token = os.getenv("HF_TOKEN")
 
-# --- 2. Load your Model from its Hub Repository using the Token ---
+# --- DEBUGGING STEP: Check if the token was found ---
+# This will print to your Space's logs.
+if auth_token is not None:
+    # Do not print the full token for security. Just confirm it was found.
+    print("✅ HF_TOKEN secret was found by the application.")
+else:
+    print("❌ HF_TOKEN secret was NOT found. Please double-check it is set in your Space Settings.")
+# --- END DEBUGGING STEP ---
+
+
+# --- 2. Load your Model using the Token ---
 MODEL_ID = "Bur3hani/karani_ofline"
+MODEL_LOADED = False
 
 try:
-    print(f"Loading tokenizer and model from Hub: {MODEL_ID}...")
-    # We pass the token to the 'from_pretrained' method.
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=auth_token)
-    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID, token=auth_token)
-    print("✅ Model and Tokenizer loaded successfully using the provided token.")
-    MODEL_LOADED = True
+    if auth_token:
+        print(f"Attempting to load tokenizer and model from Hub: {MODEL_ID}...")
+        tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=auth_token)
+        model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID, token=auth_token)
+        print("✅ Model and Tokenizer loaded successfully.")
+        MODEL_LOADED = True
+    else:
+        # This will be printed if the secret is missing.
+        print("Skipping model loading because HF_TOKEN is missing.")
+
 except Exception as e:
-    print(f"❌ Error loading from Hub: {e}")
-    MODEL_LOADED = False
+    print(f"❌ An error occurred while loading the model from the Hub: {e}")
+
 
 # --- 3. Define the Prediction Function ---
 def get_chat_response(message, history):
     if not MODEL_LOADED:
-        return "ERROR: The AI model failed to load. This could be due to a missing token or an issue with the model repository."
+        return "ERROR: The AI model failed to load. Please check the Space logs for the exact error."
 
     input_text = ""
     for user_turn, bot_turn in history: