Rausda6 commited on
Commit
caca082
·
verified ·
1 Parent(s): 2720196

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "HuggingFaceH4/zephyr-7b-alpha"
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(