Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
16 |
MAX_FILE_SIZE_MB = 20
|
17 |
MAX_FILE_SIZE_BYTES = MAX_FILE_SIZE_MB * 1024 * 1024 # Convert MB to bytes
|
18 |
|
19 |
-
MODEL_ID = "
|
20 |
|
21 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|
22 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
16 |
MAX_FILE_SIZE_MB = 20
|
17 |
MAX_FILE_SIZE_BYTES = MAX_FILE_SIZE_MB * 1024 * 1024 # Convert MB to bytes
|
18 |
|
19 |
+
MODEL_ID = "unsloth/gemma-3-1b-pt"
|
20 |
|
21 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|
22 |
model = AutoModelForCausalLM.from_pretrained(
|