Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
| 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:
|
| 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>
|