Update app.py

app.py

@@ -26,10 +26,8 @@ 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):
-  
-    return pipeline("text-generation", model=model_name, framework="tf")
-
-with warnings.catch_warnings():
+   # 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