Update app.py

app.py

@@ -25,13 +25,19 @@ user_input = st.text_area("Enter your text prompt:", "Type something here...")
 
 # Load the selected model using the pipeline
 @st.cache_resource
-def load_model(model_name):
+#def load_model(model_name):
    # return pipeline("text-generation", model=model_name, framework="tf")
+ #   with warnings.catch_warnings():
+  #  warnings.simplefilter("ignore")
+   # access_token = os.getenv("HF_ACCESS_TOKEN")  # Use an environment variable for the access token
+   # return pipeline("text-generation", model=model_name, framework="tf", use_auth_token=access_token)  # Use TensorFlow framework
+
+def load_model(model_name):
     with warnings.catch_warnings():
-    warnings.simplefilter("ignore")
+        warnings.simplefilter("ignore")
     access_token = os.getenv("HF_ACCESS_TOKEN")  # Use an environment variable for the access token
     return pipeline("text-generation", model=model_name, framework="tf", use_auth_token=access_token)  # Use TensorFlow framework
-        
+
 
 # Button to run the model
 if st.button("Generate Response"):