Sandaruth commited on
Commit
61d2a03
·
verified ·
1 Parent(s): 5290b00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ login(token=huggingface_token)
15
 
16
  # Available models for summarization
17
  models = {
18
- "T5": "Sandaruth/T5_Full_Fine_Tuned_FINDSUM",
19
  "BERT": "bert-base-uncased", # Note: BERT isn't designed for summarization; you can change this
20
  "LongT5": "google/long-t5-local-base",
21
  "Pegasus": "google/pegasus-xsum"
 
15
 
16
  # Available models for summarization
17
  models = {
18
+ "T5_Full_FineTune_V0.1_40000_Data": "ARSynopsis/T5_Full_FineTune_V0.1_40000_Data",
19
  "BERT": "bert-base-uncased", # Note: BERT isn't designed for summarization; you can change this
20
  "LongT5": "google/long-t5-local-base",
21
  "Pegasus": "google/pegasus-xsum"