MeetJivani commited on
Commit
048132e
·
1 Parent(s): 5407b63

Update summarize.py

Browse files
Files changed (1) hide show
  1. summarize.py +1 -1
summarize.py CHANGED
@@ -27,7 +27,7 @@ def load_model_and_tokenizer(model_name: str) -> tuple:
27
  "Model 4": "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
28
  "Model 5": "pszemraj/pegasus-x-large-book-summary",
29
  }
30
- selected_model_identifier = MODEL_OPTIONS.get(user_selected_model_name)
31
  device = "cuda" if torch.cuda.is_available() else "cpu"
32
  model = AutoModelForSeq2SeqLM.from_pretrained(
33
  selected_model_identifier,
 
27
  "Model 4": "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
28
  "Model 5": "pszemraj/pegasus-x-large-book-summary",
29
  }
30
+ selected_model_identifier = MODEL_OPTIONS.get(model_name)
31
  device = "cuda" if torch.cuda.is_available() else "cpu"
32
  model = AutoModelForSeq2SeqLM.from_pretrained(
33
  selected_model_identifier,