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 |
# Load once (CPU until first call; device_map will move to GPU on first run)
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
|
|
|
|
| 1 |
import os, torch, gradio as gr, spaces
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
|
| 3 |
|
| 4 |
+
MODEL_ID = os.getenv("MODEL_ID", "JDhruv14/hello")
|
| 5 |
|
| 6 |
# Load once (CPU until first call; device_map will move to GPU on first run)
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
|