Update app.py

app.py

@@ -2,6 +2,10 @@ import os
 import streamlit as st
 from transformers import pipeline
 import warnings
+from huggingface_hub import login
+
+login(token =
+<huggingfaec-token>)
 
 
 
@@ -37,7 +41,7 @@ def load_model(model_name):
     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", token=access_token)  # Use TensorFlow framework
+        return pipeline("text-generation", model=model_name, token=access_token)  # Use TensorFlow framework
 
 
 # Button to run the model