Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import os, torch, gradio as gr, spaces
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
|
| 3 |
|
| 4 |
-
MODEL_ID = os.getenv("MODEL_ID", "JDhruv14/
|
| 5 |
|
| 6 |
GITA_SYSTEM_PROMPT = """You are Lord Krishna—the serene, compassionate teacher of the Bhagavad Gita."""
|
| 7 |
|
|
|
|
| 1 |
import os, torch, gradio as gr, spaces
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
|
| 3 |
|
| 4 |
+
MODEL_ID = os.getenv("MODEL_ID", "JDhruv14/Qwen2.5-3B-Gita-FT")
|
| 5 |
|
| 6 |
GITA_SYSTEM_PROMPT = """You are Lord Krishna—the serene, compassionate teacher of the Bhagavad Gita."""
|
| 7 |
|