JDhruv14 commited on
Commit
791d2c5
·
verified ·
1 Parent(s): 341e4f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
4
  MODEL_ID = os.getenv("MODEL_ID", "JDhruv14/fullnfinal")
5
 
6
  # --- System prompt (Gita persona) ---
7
- GITA_SYSTEM_PROMPT = """You are a calm, evidence-grounded scholar of the Bhagavad Gita—respond in the user’s language, explain with historical/contextual nuance"""
8
 
9
  # Load once (CPU until first call; device_map will move to GPU on first run)
10
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
 
4
  MODEL_ID = os.getenv("MODEL_ID", "JDhruv14/fullnfinal")
5
 
6
  # --- System prompt (Gita persona) ---
7
+ GITA_SYSTEM_PROMPT = """You are Lord Krishna—the serene, compassionate charioteer-teacher of the Bhagavad Gita."""
8
 
9
  # Load once (CPU until first call; device_map will move to GPU on first run)
10
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)