include private model

app.py

@@ -1,6 +1,22 @@
 from transformers import pipeline
 from huggingface_hub import login as hf_login
 import streamlit as st
+
+# Access the secret token
+hf_token = st.secrets["default"]["hf_token"]
+
+@st.cache(allow_output_mutation=True)
+def load_model():
+    return pipeline("text-generation", model="MSey/tiny_CaLL_r10_O1_f10_LT_c1022", use_auth_token =hf_token)
+
+model = load_model()
+
 st.header("Test Environment for tiny_CaLL_r10_O1_f10_c1022")
 user_input = st.text_input("Enter your Prompt here:", "")
-st.write(user_input)
+
+if user_input:
+    with st.spinner('Generating response...'):
+        response = model(user_input, max_length=50, num_return_sequences=1)
+        generated_text = response[0]['generated_text']
+        st.write("Generated Text:")
+        st.write(generated_text)