JDhruv14 commited on
Commit
190ad71
·
verified ·
1 Parent(s): 5a30bf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 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)
@@ -136,7 +136,7 @@ with gr.Blocks(css="""
136
 
137
  gr.Markdown(
138
  """
139
- <div style='text-align: left; padding: 10px;'>
140
  <h1 style='font-size: 2.0em; margin-bottom: 0.2em;'><span style='color: #4F46E5;'>🪷 Sarathi.AI</span></h1>
141
  <p style='font-size: 1.0em; color: #bbb;'>Gita’s Eternal Teachings, Guided by AI 🕉️</p>
142
  </div>
 
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 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)
 
136
 
137
  gr.Markdown(
138
  """
139
+ <div style='text-align: center; padding: 10px;'>
140
  <h1 style='font-size: 2.0em; margin-bottom: 0.2em;'><span style='color: #4F46E5;'>🪷 Sarathi.AI</span></h1>
141
  <p style='font-size: 1.0em; color: #bbb;'>Gita’s Eternal Teachings, Guided by AI 🕉️</p>
142
  </div>