d

app.py

@@ -13,7 +13,7 @@ def main():
     st.subheader("UCLA DSU Project, Fall 2023")
     st.markdown("Daniel Mendelevitch  \n Terry Ming  \n Casey Tattersall  \n Sean Tjoa")
 
-    st.header("What Are Transformer? 🚗🔄🤖")
+    st.header("What Are Transformers? 🚗🔄🤖")
     
     header_text = """A transformer is a specific type of neural network that uses a mechanism called self-attention to learn the context (and 
         thus meaning) of sequential data. Transformer-based models can be used in many different domains, such as processing language, predicting 
@@ -70,7 +70,7 @@ def main():
     # small_model = load_casey_model(tokenizer, device)
     model = load_big_model(tokenizer, device)
 
-    st.markdown(f"Model Type: {type(model)}, Device: {device}")
+    st.markdown(print(model))
 
     if st.button('Write my story!'):
         placeholder = st.empty()