karthikmn commited on
Commit
0d89278
·
verified ·
1 Parent(s): a4e46c0

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -7,7 +7,7 @@ MODEL_NAME = "sathkrutha/deal-score-model" # Use your actual repo ID
7
  SUMMARIZER_PATH = "t5-small" # Can use distilgpt2 or other summarizers
8
 
9
  def load_model():
10
- model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
11
  tokenizer = BertTokenizer.from_pretrained(MODEL_PATH)
12
  summarizer = T5ForConditionalGeneration.from_pretrained(SUMMARIZER_PATH)
13
  return model, tokenizer, summarizer
 
7
  SUMMARIZER_PATH = "t5-small" # Can use distilgpt2 or other summarizers
8
 
9
  def load_model():
10
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
11
  tokenizer = BertTokenizer.from_pretrained(MODEL_PATH)
12
  summarizer = T5ForConditionalGeneration.from_pretrained(SUMMARIZER_PATH)
13
  return model, tokenizer, summarizer