app

app.py

@@ -12,22 +12,15 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch  
 
 
-
-
-'''
-basic line example
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
-'''
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+model_name_xsum = 'google/pegasus-xsum'
+    
+tokenizer = AutoTokenizer.from_pretrained(model_name_xsum)
+model = PegasusForConditionalGeneration.from_pretrained(model_name_xsum).to(device)
 
 
 def pegasus_abs_summarize(src_text):
         
-    device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    model_name_xsum = 'google/pegasus-xsum'
-        
-    tokenizer = AutoTokenizer.from_pretrained(model_name_xsum)
-    model = PegasusForConditionalGeneration.from_pretrained(model_name_xsum).to(device)
     
     batch = tokenizer(src_text, truncation=True, padding='longest', return_tensors="pt").to(device)