Update app.py

app.py

@@ -3,10 +3,11 @@ from groq import Groq
 import gradio as gr
 from transformers import AutoModel, AutoConfig
 
+hf_token = os.getenv("HF_TOKEN")  # Make sure you set this environment variable
+
 try:
-    config = AutoConfig.from_pretrained("HusseinEid/llama-3-chatbot", config_file_name="config.json")
-    model = AutoModel.from_pretrained("HusseinEid/lora_model", config=config)
-    
+    config = AutoConfig.from_pretrained("HusseinEid/llama-3-chatbot", config_file_name="config.json", use_auth_token=hf_token)
+    model = AutoModel.from_pretrained("HusseinEid/lora_model", config=config, use_auth_token=hf_token)
 except OSError as e:
     print(f"Error: {e}")
     
@@ -53,4 +54,4 @@ with gr. Blocks(theme=gr.themes.Monochrome(), fill_height=True) as demo:
                         )
 
 demo.queue()
-demo.launch(share=True)
+demo.launch()