Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import time
 import pandas as pd
 import numpy as np
+import torch
+from transformers import pipeline
 
 st.set_page_config(page_title="Samuel Portfolio", page_icon="📈")
 
@@ -87,5 +89,12 @@ if choice == "Plotting":
 if choice == "Contact":
     st.title("You can contact me via:")
 
+    torch.manual_seed(0)
+    generator = pipeline('text-generation', model = 'gpt2')
+    prompt = "Hello, I'm a language model"
+    
+    generator(prompt, max_length = 30)
+    st.write(generator)
+
 if choice == "Attached files": 
     st.title("Download final project report here")