Spaces:
Sleeping
Sleeping
try gemma
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ For more information on `huggingface_hub` Inference API support, please check th
|
|
| 8 |
# requires space hardware update to use large models (TODO)
|
| 9 |
# client = InferenceClient("mistralai/Mistral-Large-Instruct-2407")
|
| 10 |
# Note change in instantiation***
|
| 11 |
-
text_generator = pipeline("text-generation", model="
|
| 12 |
|
| 13 |
def respond(message, history, system_message, max_tokens, temperature, top_p):
|
| 14 |
# Construct the prompt with system message, history, and user input
|
|
|
|
| 8 |
# requires space hardware update to use large models (TODO)
|
| 9 |
# client = InferenceClient("mistralai/Mistral-Large-Instruct-2407")
|
| 10 |
# Note change in instantiation***
|
| 11 |
+
text_generator = pipeline("text-generation", model="google/gemma-2-2b")
|
| 12 |
|
| 13 |
def respond(message, history, system_message, max_tokens, temperature, top_p):
|
| 14 |
# Construct the prompt with system message, history, and user input
|