frankai98 commited on
Commit
8f925fa
·
verified ·
1 Parent(s): 9c4bbfa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def load_models():
60
  # Load the scoring model via pipeline.
61
  score_pipe = pipeline("text-classification", model="mixedbread-ai/mxbai-rerank-base-v1")
62
  # Load the Gemma text generation pipeline.
63
- gemma_pipe = pipeline("text-generation", model="google/gemma-3-1b-it", use_auth_token=hf_token)
64
  return score_pipe, gemma_pipe
65
 
66
  score_pipe, gemma_pipe = load_models()
 
60
  # Load the scoring model via pipeline.
61
  score_pipe = pipeline("text-classification", model="mixedbread-ai/mxbai-rerank-base-v1")
62
  # Load the Gemma text generation pipeline.
63
+ gemma_pipe = pipeline("text-generation", model="google/gemma-3-1b-it")
64
  return score_pipe, gemma_pipe
65
 
66
  score_pipe, gemma_pipe = load_models()