Spaces:
Sleeping
Sleeping
Update tools/news_reporter.py
Browse files- tools/news_reporter.py +2 -2
tools/news_reporter.py
CHANGED
@@ -3,8 +3,8 @@ from agent_state import AgentState
|
|
3 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
4 |
import os
|
5 |
|
6 |
-
# MODEL_ID = "google/gemma-3n-E4B-it" #
|
7 |
-
MODEL_ID = 'google/gemma-2b'
|
8 |
|
9 |
|
10 |
# Optional: use a cache directory
|
|
|
3 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
4 |
import os
|
5 |
|
6 |
+
# MODEL_ID = "google/gemma-3n-E4B-it" # out of memory
|
7 |
+
MODEL_ID = 'google/gemma-2-2b-it'
|
8 |
|
9 |
|
10 |
# Optional: use a cache directory
|