thedeba commited on
Commit
50cfe8b
·
verified ·
1 Parent(s): bff4f6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model_name = "thedeba/debai" # HF Hub model path
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForCausalLM.from_pretrained(model_name)
12
  device = "cpu" # Spaces free tier uses CPU; you can switch to "cuda" if GPU granted
13
- model.to(device)
14
 
15
  # -------------------------------
16
  # FastAPI setup
 
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForCausalLM.from_pretrained(model_name)
12
  device = "cpu" # Spaces free tier uses CPU; you can switch to "cuda" if GPU granted
13
+ #model.to(device)
14
 
15
  # -------------------------------
16
  # FastAPI setup