Hugging Face's logo

Spaces:
newtechdevng
/
ai_math_assistant
Sleeping

App Files Community
newtechdevng commited on
Commit
bbe47c3
·
verified ·
1 Parent(s): ea41a05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,13 +4,11 @@ import torch
4
 
5
  app = Flask(__name__)
6
 
7
- MODEL_ID = "newtechdevng/math-tutor-smollm2-360M" # full model, load directly
8
-
9
  SYSTEM_PROMPT = "You are a helpful math assistant."
10
 
11
  print("Loading model...")
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
13
-
14
  model = AutoModelForCausalLM.from_pretrained(
15
  MODEL_ID,
16
  dtype=torch.float32,
 
4
 
5
  app = Flask(__name__)
6
 
7
+ MODEL_ID = "newtechdevng/math-tutor-smollm2-360M"
 
8
  SYSTEM_PROMPT = "You are a helpful math assistant."
9
 
10
  print("Loading model...")
11
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
  MODEL_ID,
14
  dtype=torch.float32,