Barat123 commited on
Commit
1f9dfa0
·
verified ·
1 Parent(s): 14bb5cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from huggingface_hub import login
6
 
7
  login(token=os.getenv("hf_tok"))
8
 
9
- MODEL_NAME = "google/gemma-3-2b-it"
10
  generator = pipeline(
11
  "text-generation",
12
  model=MODEL_NAME,
 
6
 
7
  login(token=os.getenv("hf_tok"))
8
 
9
+ MODEL_NAME = "google/gemma-2b-it"
10
  generator = pipeline(
11
  "text-generation",
12
  model=MODEL_NAME,