Update app.py
Browse files
app.py
CHANGED
@@ -9,10 +9,6 @@ os.environ["KERAS_BACKEND"] = "jax" # Or "torch" or "tensorflow".
|
|
9 |
|
10 |
os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
|
11 |
|
12 |
-
# Load your Hugging Face model and tokenizer
|
13 |
-
model_name = "soufyane/gemma_data_science"
|
14 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
15 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
16 |
|
17 |
model = keras_nlp.models.CausalLM.from_preset(f"hf://soufyane/gemma_data_science")
|
18 |
|
|
|
9 |
|
10 |
os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
|
11 |
|
|
|
|
|
|
|
|
|
12 |
|
13 |
model = keras_nlp.models.CausalLM.from_preset(f"hf://soufyane/gemma_data_science")
|
14 |
|