Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|
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()
|