Update app.py

app.py

@@ -1,5 +1,6 @@
 from transformers import TextClassificationPipeline, AutoTokenizer, AutoModelForSequenceClassification
 import gradio as gr
+import os
 
 # def get_model(model_name='Overfit-GM/temp_dist'):
 #     id2label = {0: 'INSULT', 1: 'OTHER',
@@ -22,8 +23,8 @@ models = [
 ]
 
 model_box=[
-    gr.load(models[0], src='models'),
-    gr.load(models[1], src='models'),
+    gr.load(models[0], src='models', hf_token=os.environ['API_KEY']),
+    gr.load(models[1], src='models', hf_token=os.environ['API_KEY']),
 ]
 
 def sentiment_analysis(text, model_choice):