Hugging Face's logo

Spaces:
ashbeexd
/
emotion-classifier
Sleeping

App Files Community
Ashwin B commited on
Commit
e7784a7
·
1 Parent(s): d5e81f8

app fixes

Browse files
Files changed (1) hide show
  1. app/app.py +2 -1
app/app.py CHANGED
@@ -21,7 +21,8 @@ GOEMOTIONS_LABELS = [
21
  # Load model and tokenizer
22
  # -----------------------------
23
  MODEL_PATH = "outputs/model"
24
- model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH, device_map="auto", low_cpu_mem_usage=True)
 
25
  tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
26
  print("Model device:", next(model.parameters()).device)
27
 
 
21
  # Load model and tokenizer
22
  # -----------------------------
23
  MODEL_PATH = "outputs/model"
24
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
25
+ model.to("cpu")
26
  tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
27
  print("Model device:", next(model.parameters()).device)
28