Ashwin B commited on
Commit
d5e81f8
·
1 Parent(s): 2d1282b

Fixing model path

Browse files
Files changed (1) hide show
  1. app/app.py +1 -1
app/app.py CHANGED
@@ -20,7 +20,7 @@ GOEMOTIONS_LABELS = [
20
  # -----------------------------
21
  # Load model and tokenizer
22
  # -----------------------------
23
- MODEL_PATH = "/mount/src/emotion-classifier-nlp/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)
 
20
  # -----------------------------
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)