DarkMindForever commited on
Commit
9b20c73
·
verified ·
1 Parent(s): 391a834

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ if HF_TOKEN:
15
  login(token=HF_TOKEN)
16
 
17
  # --- 2. MODEL SETUP ---
18
- MODEL_ID = "google/gemma-2-2b-it"
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
21
  model = AutoModelForCausalLM.from_pretrained(
 
15
  login(token=HF_TOKEN)
16
 
17
  # --- 2. MODEL SETUP ---
18
+ MODEL_ID = "openai-community/gpt2-xl"
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
21
  model = AutoModelForCausalLM.from_pretrained(