Update app.py

app.py

@@ -1,5 +1,13 @@
 import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+from transformers import AutoTokenizer, FalconModel
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("Rocketknight1/falcon-rw-1b")
+model = FalconModel.from_pretrained("Rocketknight1/falcon-rw-1b")
+
+inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
+outputs = model(**inputs)
+
+last_hidden_states = outputs.last_hidden_state