rui3000 commited on
Commit
fce68ad
·
verified ·
1 Parent(s): badf23e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import spaces # Import the spaces library
5
  from transformers import AutoModelForCausalLM, AutoTokenizer
6
 
7
  # --- Configuration ---
8
- MODEL_ID = "Qwen/Qwen2-1.5B-Math" # Replace with actual ID if found
9
  # --- Load Model and Tokenizer ---
10
  print(f"Loading model: {MODEL_ID}")
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
5
  from transformers import AutoModelForCausalLM, AutoTokenizer
6
 
7
  # --- Configuration ---
8
+ MODEL_ID = "Qwen/Qwen2.5-Math-1.5B" # Replace with actual ID if found
9
  # --- Load Model and Tokenizer ---
10
  print(f"Loading model: {MODEL_ID}")
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)