Update app.py

app.py

@@ -5,10 +5,10 @@ import streamlit as st
 def context_text(text): return f"### Context\n{text}\n\n### Answer"
 
 @st.cache_resource
-def load_model():
+def load_pipe():
     return pipeline("text-generation", model="MSey/tiny_CaLL_r10_O1_f10_LT_c1022")
 
-model = load_model()
+pipe = load_pipe()
 
 st.header("Test Environment for tiny_CaLL_r10_O1_f10_LT_c1022")
 user_input = st.text_input("Enter your Prompt here:", "")
@@ -17,7 +17,7 @@ context_len = len(contexted_ipnut)
 
 if user_input:
     with st.spinner('Generating response...'):
-        response = model(contexted_ipnut, max_new_tokens = 200, num_return_sequences=1)
+        response = pipe(contexted_ipnut, max_new_tokens = 200, num_return_sequences=1)
         generated_text = response[0]['generated_text'][context_len:]
         st.write("Generated Text:")
         st.markdown(generated_text)