Spaces:

ARSynopsis
/

Text_Summarization

Runtime error

Sandaruth commited on Oct 3, 2024

Commit

f2af6e2

verified ·

1 Parent(s): 61d2a03

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,8 @@ login(token=huggingface_token)
 # Available models for summarization
 models = {
-    "T5_Full_FineTune_V0.1_40000_Data": "ARSynopsis/T5_Full_FineTune_V0.1_40000_Data",
     "BERT": "bert-base-uncased",  # Note: BERT isn't designed for summarization; you can change this
     "LongT5": "google/long-t5-local-base",
     "Pegasus": "google/pegasus-xsum"

 # Available models for summarization
 models = {
+    "T5_Full_FineTune_V0.1_40K": "ARSynopsis/T5_Full_FineTune_V0.1_40K",
+    "T5_Full_FineTune_V0.1_80K": "ARSynopsis/T5_Full_FineTune_V0.1_80K",
     "BERT": "bert-base-uncased",  # Note: BERT isn't designed for summarization; you can change this
     "LongT5": "google/long-t5-local-base",
     "Pegasus": "google/pegasus-xsum"