abs sum

app.py

@@ -13,13 +13,33 @@ import torch
 
 
 
+
+'''
+basic line example
 x = st.slider('Select a value')
+st.write(x, 'squared is', x * x)
+'''
 
 
-st.text_input("Your name", key="name")
+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)
+    
+    translated = model.generate(**batch)
+    
+    target_text = tokenizer.batch_decode(translated, skip_special_tokens =True)
+    
+    return target_text
 
-st.write(x, 'squared is', x * x)
 
+st.text_input("Input:", key="input")
 
+abs_output = pegasus_abs_summarize(st.session_state.input)
 
-st.write('name', st.session_state.name)
+st.write('Abstractive output', abs_output)